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

[CUDA] Implement SegmentedMM (#3238)

Browse Source
This commit is contained in:
Long Yixing
2026-03-11 13:31:43 -07:00
committed by GitHub
parent 1c2d7041ab
commit a9573f92f6
7 changed files with 399 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/tests/cuda_skip.py
-2
View File
@@ -6,8 +6,6 @@ cuda_skip = {
"TestBlas.test_gather_matmul",
"TestBlas.test_gather_matmul_grad",
"TestBlas.test_gather_mm_sorted_vjp",
# Segmented matmul NYI
"TestBlas.test_segmented_mm",
# FFTs NYI
"TestFFT.test_fft",
"TestFFT.test_fft_big_powers_of_two",