search

teddykoker / torchsort

Fast, differentiable sorting and ranking in PyTorch
https://pypi.org/project/torchsort/
Apache License 2.0
765 stars 33 forks source link

Better error message if CUDA extension is not available #43

Closed teddykoker closed 2 years ago

teddykoker commented 2 years ago

Addresses unhelpful error if attempting to use CUDA extension when it has not been built (seen in #33)