L-electron-Rare/mlx
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[CUDA] Add FFT support (#3243)

Browse Source
This commit is contained in:
Lucas Newman
2026-03-14 21:02:19 +09:00
committed by GitHub
parent b0564a9112
commit 5d1700493a
9 changed files with 498 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/tests/cuda_skip.py
-10
View File
@@ -6,16 +6,6 @@ cuda_skip = {
"TestBlas.test_gather_matmul",
"TestBlas.test_gather_matmul_grad",
"TestBlas.test_gather_mm_sorted_vjp",
# FFTs NYI
"TestFFT.test_fft",
"TestFFT.test_fft_big_powers_of_two",
"TestFFT.test_fft_contiguity",
"TestFFT.test_fft_exhaustive",
"TestFFT.test_fft_grads",
"TestFFT.test_fft_into_ifft",
"TestFFT.test_fft_large_numbers",
"TestFFT.test_fft_shared_mem",
"TestFFT.test_fftn",
# Lapack ops NYI
"TestLinalg.test_cholesky",
"TestLinalg.test_cholesky_inv",