adaptive_conv_cuda_impl adaptive_conv_cuda.cpp adaptive_conv_kernel.cu

[yiiizuo]

How can I replace the CUDA compilation modules in the code within Ascend?

code link: https://github.com/mhamilton723/FeatUp
pip install git+https://github.com/mhamilton723/FeatUp

#include <cuda.h>
#include <ATen/cuda/CUDAContext.h>
#include <cuda_runtime.h>

ext_modules=[
CUDAExtension(
'adaptive_conv_cuda_impl',
[
'featup/adaptive_conv_cuda/adaptive_conv_cuda.cpp',
'featup/adaptive_conv_cuda/adaptive_conv_kernel.cu',
]),
CppExtension(
'adaptive_conv_cpp_impl',
['featup/adaptive_conv_cuda/adaptive_conv.cpp'],
undef_macros=["NDEBUG"]),
],