PyTorch-extension-Convolution
PyTorch-extension-Convolution copied to clipboard
huangtinglin
An example of C++ extension for PyTorch.
Results
1
PyTorch-extension-Convolution issues
Sort by
recently updated
recently updated
newest added
trafficstars
The definitions at the top of conv_cuda.cpp should be changed to avoid errors: #define CHECK_CUDA(x) AT_ASSERTM(x.type().is_cuda(), #x " must be a CUDA tensor") #define CHECK_CONTIGUOUS(x) AT_ASSERTM(x.is_contiguous(), #x " must be...
