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

[CUDA] Handle residue k in qmm_naive (#3379)

Browse Source
This commit is contained in:
Cheng
2026-04-18 13:30:07 +09:00
committed by GitHub
parent 859f22fbb0
commit fa4320d5fa
7 changed files with 147 additions and 79 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/tests/cuda_skip.py
-7
View File
@@ -19,11 +19,4 @@ cuda_skip = {
"TestQuantized.test_gather_qmm",
"TestQuantized.test_gather_qmm_sorted",
"TestQuantized.test_gather_qmm_grad",
"TestQuantized.test_non_multiples",
"TestQuantized.test_qmm_shapes",
"TestQuantized.test_fp_qvm",
"TestQuantized.test_qvm",
"TestQuantized.test_qmv_small_non_multiples",
"TestQuantized.test_small_matrix",
"TestExportImport.test_export_quantized_model",
}