Hi all,
I’m trying to modify ‘pytorch/aten/src/ATen/cudnn/Conv.cpp’ to add some customized functions with cuda kernel.
Just wondering what’s the relations between ‘at::Tensor’ and ‘THCTensor’, ‘THCudaTensor’. Sine ‘Conv.cpp’ mainly uses ‘at::Tensor’, but my customized .cu file use ‘THCudaTensor’.
How can I convert ‘at::Tensor’ to ‘THCudaTensor’ at Conv.cpp and call my customized cuda kernel?
Your help will be greatly appreciated, thanks!