it is a template to learn how to transfer libtorch tensor to a custom cuda kernel
mkdir build
cd build
cmake ..
make
./cuda_libtorch_test-
Notifications
You must be signed in to change notification settings - Fork 0
zwshan/libtorch_with_cuda_kernel
About
libtorch with custom cuda kernel
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published