search

rusty1s / pytorch_sparse

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

Fix `cudaSetDevice` for CUDA 12 #370

Closed rusty1s closed 6 months ago