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

[CUDA] Implement gather_mm_rhs (#2902)

Browse Source
This commit is contained in:
Cheng
2025-12-24 09:42:56 +09:00
committed by GitHub
parent 1eef1d155c
commit 1d21d0e696
7 changed files with 458 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/tests/cuda_skip.py
+1 -1
View File
@@ -6,7 +6,7 @@ cuda_skip = {
# Gather matmul NYI
"TestBlas.test_gather_matmul",
"TestBlas.test_gather_matmul_grad",
"TestBlas.test_gather_mm_sorted",
"TestBlas.test_gather_mm_sorted_vjp",
# Segmented matmul NYI
"TestBlas.test_segmented_mm",
# Hadamard NYI