Point-M2AE icon indicating copy to clipboard operation
Point-M2AE copied to clipboard

question about the pytorch version

Open youthfrog opened this issue 2 years ago • 2 comments

I'm wondering your pytorch version. My torch vision is 2.0.0. It seems that this version will have problem about the complication of the emd_kernal.cu because of the disappearance of the THC/THC.h. Looking forward to your reply.

youthfrog avatar Oct 04 '23 11:10 youthfrog

He might use a Pytorch version lower than 1.8 because THCuda was deprecated since. If you want to use the newest pytorch version you can:

  • Remove the #include <THC/THC.h> since it's deprecated (https://github.com/pytorch/pytorch/pull/66391) and replace all THCudaCheck by C10_CUDA_CHECK.
  • Change all the CHECK_EQ to TORCH_CHECK_EQ since the alias name as changed (https://github.com/pytorch/pytorch/pull/82032)

Have fun 🤩

julien-blanchon avatar Nov 17 '23 16:11 julien-blanchon

@ZrrSkywalker If you want I can make a PR to fix this, let me know

julien-blanchon avatar Nov 17 '23 16:11 julien-blanchon