error: identifier "AT_CHECK" is undefined

Wangt-CN / VC-R-CNN

[CVPR 2020] The official pytorch implementation of ``Visual Commonsense R-CNN''

MIT License

352 stars 61 forks source link

error: identifier "AT_CHECK" is undefined #12

Open Abonnen opened 4 years ago

Abonnen commented 4 years ago

I followed the instructions to install VC-R-CNN, however when I run the last command python setup.py build develop in install.sh, I got errors:

error: identifier "AT_CHECK" is undefined

I tried the solution in another issue , I replaced AT_CHECK by TORCH_CHECK in file vc_rcnn/csrc/cuda/deform_conv_cuda.cu and vc_rcnn/csrc/cuda/deform_pool_cuda.cu and it worked.

Actually, I don't know why this happened. If anyone get the same error, I hope it helps.

ArghyaPal commented 3 years ago

add

#ifndf AT_CHECK
#define AT_CHECK TORCH_CHECK
#endif

at the very beginning of:

vc_rcnn/csrc/cuda/deform_conv_cuda.cu

and

vc_rcnn/csrc/cuda/deform_pool_cuda.cu