├── .gitignore ├── CMakeLists.txt ├── README.md ├── lltm ├── __init__.py ├── lltm.cpp ├── lltm.py └── lltm_cuda.cu ├── requirements.txt └── setup.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cmpute/pytorch-cmake-example/HEAD/.gitignore -------------------------------------------------------------------------------- /CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cmpute/pytorch-cmake-example/HEAD/CMakeLists.txt -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cmpute/pytorch-cmake-example/HEAD/README.md -------------------------------------------------------------------------------- /lltm/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cmpute/pytorch-cmake-example/HEAD/lltm/__init__.py -------------------------------------------------------------------------------- /lltm/lltm.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cmpute/pytorch-cmake-example/HEAD/lltm/lltm.cpp -------------------------------------------------------------------------------- /lltm/lltm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cmpute/pytorch-cmake-example/HEAD/lltm/lltm.py -------------------------------------------------------------------------------- /lltm/lltm_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cmpute/pytorch-cmake-example/HEAD/lltm/lltm_cuda.cu -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cmpute/pytorch-cmake-example/HEAD/setup.py --------------------------------------------------------------------------------