华为计算微信公众号
昇腾AI开发者公众号
华为计算微博
华为计算今日头条
from torch.utils.cpp_extension import CUDA_HOME
from torch.utils.cpp_extension import CppExtension
from torch.utils.cpp_extension import CUDAExtension
模型中有用到这些,迁移到npu后,这部分代码会报错,该怎么把cuda的模型迁移到npu上后解决这部分问题,以及cuda算子该怎么适配
我要发帖子
from torch.utils.cpp_extension import CUDA_HOME
from torch.utils.cpp_extension import CppExtension
from torch.utils.cpp_extension import CUDAExtension
模型中有用到这些,迁移到npu后,这部分代码会报错,该怎么把cuda的模型迁移到npu上后解决这部分问题,以及cuda算子该怎么适配