Nvmath-Python: Nvidia Math Libraries for the Python Ecosystem (opens in new tab)

(github.com)

67 pointsgballan6mo ago4 comments

4 comments

Closer to arithmetic libraries than math libraries.

Nice, custom prologues and epilogues for FFTs are a welcome feature.

That stuff was in FFTW and cuFFT for many many years. It's one of the features I always pointed to as missing in rocFFT but they've finally caught up.

KeplerBoy6mo ago

Yes, it has been in cuFFT for a long time. I thought it's new to python cuda libraries, but no. Turns out it has also been a thing in CuPy for a long time.

j / k navigate · click thread line to collapse

4 comments

amelius6mo ago

Closer to arithmetic libraries than math libraries.

KeplerBoy6mo ago

Nice, custom prologues and epilogues for FFTs are a welcome feature.

physicsguy6mo ago

That stuff was in FFTW and cuFFT for many many years. It's one of the features I always pointed to as missing in rocFFT but they've finally caught up.

KeplerBoy6mo ago

Yes, it has been in cuFFT for a long time. I thought it's new to python cuda libraries, but no. Turns out it has also been a thing in CuPy for a long time.

j / k navigate · click thread line to collapse