Install emd extension with torch>1.11.0

Pang-Yatian / Point-MAE

[ECCV2022] Masked Autoencoders for Point Cloud Self-supervised Learning

MIT License

448 stars 55 forks source link

Closed lqxisok closed 9 months ago

lqxisok commented 9 months ago

extensions/emd/cuda/emd_kernel.cu
- comment #include <THC/THC.h>
- replace all CHECK_EQ to TORCH_CHECK_EQ
- replace all THCudaCheck to AT_CUDA_CHECK

X-Bruce-Y commented 2 months ago

life saver, man