sigsep-mus-eval icon indicating copy to clipboard operation
sigsep-mus-eval copied to clipboard

Support optional cupy for faster BSS evaluation

Open sevagh opened this issue 4 years ago • 1 comments

Speed up BSS metrics with the GPU and CUDA-Python (cupy).

Regression tests pass. Usability is possibly a concern, given that clear_cupy_cache() should be run regularly between tracks to avoid GPU OOM errors when the FFT plan cache fills up for a single evaluated track.

I can squash my git commits as required.

sevagh avatar Aug 19 '21 13:08 sevagh

In hindsight I'm kinda uncertain where the "disable_cupy", "clear_cupy_cache", etc. functions would go, or if they even belong in the package.

sevagh avatar Sep 04 '21 10:09 sevagh