-
Notifications
You must be signed in to change notification settings - Fork 2
Open
Description
更改主动旋转滤波器的cuda源码,将#include <THC/THC.h>注释掉,把THCudaCheck替换为AT_CUDA_CHECK,并替换THCCeilDiv(x,y)为(x+y-1)/y,完成以上操作之前我试了一下再次运行python setup.py build_ext --inplace观察是否为原本的THC问题,但是报出错误变成了IndexError: list index out of range,完成THC更改的操作后,运行同样出现了这个错误,使用了python setup.py clean命令再运行仍然报错,部分报错代码如下File "/home/mjh/anaconda3/lib/python3.9/site-packages/torch/utils/cpp_extension.py", line 1694, in _get_cuda_arch_flags
arch_list[-1] += '+PTX'
IndexError: list index out of range
麻烦作者大大了,感谢感谢
Metadata
Metadata
Assignees
Labels
No labels