Open Divyam10 opened 10 months ago
โน๏ธ Ensure Compatibility with Newer PyTorch and CUDA Versions for RANSAC Voting Layer
๐ Summary: This commit updates the RANSAC voting module to ensure compatibility with the latest PyTorch and CUDA versions.
๐ Changes Made:
Thc/thc.h
extern THCState* state;
ransac_voting.cpp
torch.solve()
torch.linalg.solve()
tensor.ge()
tensor.masked_select
๐ Additional Notes:
๐ง Dependencies:
๐งช Testing:
๐ Acknowledgments: Thanks to the authors for sharing code for this valuable research.
๐ Date: 21st Jan, 2024
โน๏ธ Ensure Compatibility with Newer PyTorch and CUDA Versions for RANSAC Voting Layer
๐ Summary: This commit updates the RANSAC voting module to ensure compatibility with the latest PyTorch and CUDA versions.
๐ Changes Made:
Thc/thc.h
headers from version >11.1 which gives compilation errors. Commentedextern THCState* state;
inransac_voting.cpp
torch.solve()
withtorch.linalg.solve()
for newer version.tensor.ge()
function on the mask astensor.masked_select
expects a boolean tensor array in newer versions.๐ Additional Notes:
๐ง Dependencies:
๐งช Testing:
๐ Acknowledgments: Thanks to the authors for sharing code for this valuable research.
๐ Date: 21st Jan, 2024