Problems installing the package

[tacchinotacchi]

While trying to install this package, I've had a couple problems:

1. I had PyTorch installed through conda, pip was trying to install this torch version in an isolated env to build the package, couldn't find it and was hanging in an infinite loop. Solved with --no-build-isolation, wonder if there's a way to do this in the repo files instead of relying on the user?
2. This exception is thrown when trying to import the package:

FileNotFoundError                         Traceback (most recent call last)
Input In [2], in <module>
----> 1 import torch_discounted_cumsum

# uninteresting stack trace...

FileNotFoundError: [Errno 2] No such file or directory: '.../lib/python3.9/site-packages/torch_discounted_cumsum/discounted_cumsum_cuda.cpp'

it seems the cuda source file is not kept around in the installed package, I'm not sure why?

[tacchinotacchi]

My bad, the second problem was because I wasn't setting CUDA_HOME during the initial installation.