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

[CUDA] gather_mm (#3414)

Browse Source
This commit is contained in:
Cheng
2026-04-17 16:53:44 +09:00
committed by GitHub
parent 940ba473fe
commit d142de6a20
7 changed files with 399 additions and 69 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/tests/cuda_skip.py
-4
View File
@@ -1,8 +1,4 @@
cuda_skip = {
# Gather matmul NYI
"TestBlas.test_gather_matmul",
"TestBlas.test_gather_matmul_grad",
"TestBlas.test_gather_mm_sorted_vjp",
# Lapack ops NYI
"TestLinalg.test_cholesky",
"TestLinalg.test_cholesky_inv",