├── .gitignore ├── CMakeLists.txt ├── LICENSE ├── README.md ├── main.cu └── test.py /.gitignore: -------------------------------------------------------------------------------- 1 | build/ 2 | -------------------------------------------------------------------------------- /CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/r-barnes/pytorch_cmake_example/HEAD/CMakeLists.txt -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/r-barnes/pytorch_cmake_example/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/r-barnes/pytorch_cmake_example/HEAD/README.md -------------------------------------------------------------------------------- /main.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/r-barnes/pytorch_cmake_example/HEAD/main.cu -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/r-barnes/pytorch_cmake_example/HEAD/test.py --------------------------------------------------------------------------------