zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-12-16 01:49:05 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Metal FFT for powers of 2 up to 2048 (#915)

Browse Source 
* 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>
This commit is contained in:
Alex Barron
2024-04-12 05:40:06 +01:00
committed by GitHub
parent ae18326533
commit 2e7c02d5cd
6 changed files with 431 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
mlx/backend/metal/kernels/CMakeLists.txt
View File

@@ -21,6 +21,7 @@ set(
"binary_two"
"conv"
"copy"
"fft"
"gemv"
"quantized"
"random"