GitHub - kalam360/cuda_example_pytorch
github.com › kalam360 › cuda_example_pytorchSeveral simple examples for neural network toolkits (PyTorch, TensorFlow, etc.) calling custom CUDA operators. We provide several ways to compile the CUDA kernels and their cpp wrappers, including jit, setuptools and cmake. We also provide several python codes to call the CUDA kernels, including kernel time statistics and model training.
GitHub - kalam360/cuda_example_pytorch
https://github.com/kalam360/cuda_example_pytorch15.12.2021 · Neural Network CUDA Example. Several simple examples for neural network toolkits (PyTorch, TensorFlow, etc.) calling custom CUDA operators. We provide several ways to compile the CUDA kernels and their cpp wrappers, including jit, setuptools and cmake. We also provide several python codes to call the CUDA kernels, including kernel time ...