最近跑pytorch项目的时候出了个问题
要运行build.sh文件才能生成deform_conv_cuda文件
结果一大堆error: identifier "AT_CHECK" is undefined
google了之后找到了解决方法:
把deform_conv_cuda.cpp里面所有的AT_CHECK替换为TORCH_CHECK
或者宏定义
#ifndef AT_CHECK
#define AT_CHECK TORCH_CHECK
#endif
最近跑pytorch项目的时候出了个问题
要运行build.sh文件才能生成deform_conv_cuda文件
结果一大堆error: identifier "AT_CHECK" is undefined
google了之后找到了解决方法:
把deform_conv_cuda.cpp里面所有的AT_CHECK替换为TORCH_CHECK
或者宏定义
#ifndef AT_CHECK
#define AT_CHECK TORCH_CHECK
#endif