rusty1s / pytorch_sparse

PyTorch Extension Library of Optimized Autograd Sparse Matrix Operations
MIT License
1.01k stars 147 forks source link

Fix compilation errors occurring when building with PyTorch-nightly #305

Closed DamianSzwichtenberg closed 1 year ago

DamianSzwichtenberg commented 1 year ago

Fix compilation errors occurring when building with PyTorch-nightly. Wrapping the cuda_version function with a lambda helps to deduce template type.

Related PR: pytorch_scatter::PR-347