Won't build with PyTorch 1.11

[igormp]

The current code won't build with PyTorch 1.11, since it seems that they removed the TH/THC include files, as per this comment: open-mmlab/mmdetection3d#1332 (comment)

The end result is this:

AdelaiDet/adet/layers/csrc/ml_nms/ml_nms.cu:4:10: fatal error: THC/THC.h: No such file or directory
    4 | #include <THC/THC.h>
      |          ^~~~~~~~~~~
compilation terminated.
ninja: build stopped: subcommand failed.

[igormp]

Seems like #518 fixes this issue.

[vanga]

I have this working on Colab with pytorch 1.11. But, I get this error on Linux, A100 with pytorch being the same version.

torch                   1.11.0+cu113
torchaudio              0.11.0+cu113
torchvision             0.12.0+cu113

/AdelaiDet/adet/layers/csrc/ml_nms/ml_nms.cu:4:10: fatal error: THC/THC.h: No such file or directory
 #include <THC/THC.h>
          ^~~~~~~~~~~
compilation terminated.
error: command '/usr/local/cuda-11.3/bin/nvcc' failed with exit code 1

is it related to cuda package then?

[vanga]

I got it working with the change from the PR #518 and also removing line 4 as mentioned in the PR comments

AdelaiDet/adet/layers/csrc/ml_nms/ml_nms.cu

Line 4 in efab1f5

#include <THC/THC.h>

[visionKinger]

I got it working with the change from the PR #518 and also removing line 4 as mentioned in the PR comments

AdelaiDet/adet/layers/csrc/ml_nms/ml_nms.cu

Line 4 in efab1f5

#include <THC/THC.h>

after removing the line 4,more error happened

[2605759123]

I got it working with the change from the PR #518 and also removing line 4 as mentioned in the PR comments

AdelaiDet/adet/layers/csrc/ml_nms/ml_nms.cu

Line 4 in efab1f5

#include <THC/THC.h>

It works. Perfect!