/pytorch-cmake-example

Example to build PyTorch CUDA extension using CMake (with pybind11 and scikit-build)

Primary LanguageCuda

Watchers