Relation between 'at::Tensor', 'THCTensor' and 'THCudaTensor'

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!

8 Likes