search

FFTW / fftw3

DO NOT CHECK OUT THESE FILES FROM GITHUB UNLESS YOU KNOW WHAT YOU ARE DOING. (See below.)
GNU General Public License v2.0
2.66k stars 651 forks source link

Support fp16 and bf16 precision #329

Open zhuohoudeputao opened 11 months ago

zhuohoudeputao commented 11 months ago

Since increasingly deep learning or ai related computation relies on FFT and calculates half precision, is there any plan to support fp16 and bf16 precision for fftw?

stevengj commented 11 months ago

It should be straightforward to add, in principle. The scalar case would be trivial to add, while SIMD support will take a bit more work.

zhuohoudeputao commented 11 months ago

libbench should also support testing for half precision, too.