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

Feature complete Metal FFT (#1102)

Browse Source 
* feature complete metal fft

* fix contiguity bug

* jit fft

* simplify rader/bluestein constant computation

* remove kernel/utils.h dep

* remove bf16.h dep

* format

---------

Co-authored-by: Alex Barron <abarron22@apple.com>
This commit is contained in:
Alex Barron
2024-06-06 12:57:25 -07:00
committed by GitHub
parent 0e585b4409
commit 27d70c7d9d
17 changed files with 2601 additions and 367 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -48,6 +48,9 @@ set(
binary.h
ternary.h
copy.h
fft.h
fft/radix.h
fft/readwrite.h
softmax.h
sort.h
scan.h