search

sshaoshuai / Pointnet2.PyTorch

A faster implementation of PointNet++ based on PyTorch.
MIT License
474 stars 107 forks source link

Add PyTorch1.2+ support #25

Open keineahnung2345 opened 3 years ago

keineahnung2345 commented 3 years ago
  1. AT_CHECK -> TORCH_CHECK, ref: https://github.com/mrlooi/rotated_maskrcnn/issues/31
  2. THCState_getCurrentStream -> c10::cuda::getCurrentCUDAStream, ref: https://github.com/pytorch/pytorch/commit/a5f01846c2bae0d78a5c0cd74ced20a07b8ef961#diff-1dcab1425fcb102bc8786f370fa99028