fix compilation for higher version of pytorch.

cvg / DeepLSD

Implementation of the paper "DeepLSD: Line Segment Detection and Refinement with Deep Image Gradients"

MIT License

486 stars 62 forks source link

fix compilation for higher version of pytorch. #11

Closed B1ueber2y closed 1 year ago

B1ueber2y commented 1 year ago

THC/THC.h is removed in the new version of pytorch + cuda (refer to https://discuss.pytorch.org/t/question-about-thc-thc-h/147145)

In this pull request the function THCudaCheck() is substituted with the new interface AT_CUDA_CHECK.

Test succeeded with torch 1.11.0+cu113. Maybe it is still needed to test under torch 1.10.0 to see if it works with the minimum required version in the requirements.txt

rpautrat commented 1 year ago

I actually upgraded to torch 1.12, for future compatibility. Better to start with the latest version of torch.