mlx/python
Alex Barron 2e7c02d5cd
Metal FFT for powers of 2 up to 2048 (#915)
* add Metal FFT for powers of 2

* skip GPU test on linux

* fix contiguity bug

* address comments

* Update mlx/backend/metal/fft.cpp

* Update mlx/backend/metal/fft.cpp

* fix bug in synch

---------

Co-authored-by: Alex Barron <abarron22@apple.com>
Co-authored-by: Awni Hannun <awni.hannun@gmail.com>
Co-authored-by: Awni Hannun <awni@apple.com>
2024-04-11 21:40:06 -07:00
..
mlx Upsample with bicubic interpolation (#967) 2024-04-10 15:47:22 -07:00
src Explicit barriers with concurrent dispatch (#977) 2024-04-10 21:45:31 -07:00
tests Metal FFT for powers of 2 up to 2048 (#915) 2024-04-11 21:40:06 -07:00