Run on Pytorch 1.7.0

[weijiawang96]

Dear Authors,

Thank you very much for sharing your code. As the minimum Pytorch version I can use is 1.7.0, I wish to know any ways to run your code on it.

If I directly run python setup.py install
It says RuntimeError: Error compiling objects for extension
Which is caused by: failed to build xxx/PRA-Net-master/_ext-src/build/temp.linux-x86_64-3.7/src/sampling.o (and ball_query etc.)

I also tried adding the following 3 lines in every .h and .cu file:
#ifndef AT_CHECK
#define AT_CHECK TORCH_CHECK
#endif

But still the same error.
Is there any other solution? Thank you in advance!

[weijiawang96]

I followed quan-dao's solution here: sshaoshuai/Pointnet2.PyTorch#26 and the cpp files are built. Now I can use them. Thanks