zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-10-18 07:18:29 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

[CUDA] Affine quantize (#2354)

Browse Source 
* affine quantize and dequantize kernels

* format

* fix

* format
This commit is contained in:
Awni Hannun
2025-07-14 15:45:44 -07:00
committed by GitHub
parent e569803d7c
commit e7d2ebadd2
4 changed files with 384 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
python/tests/cuda_skip.py
View File

@@ -83,7 +83,6 @@ cuda_skip = {
"TestQuantized.test_qmm_shapes",
"TestQuantized.test_qmm_vjp",
"TestQuantized.test_qmv",
"TestQuantized.test_quantize_dequantize",
"TestQuantized.test_qvm",
"TestQuantized.test_qvm_splitk",
"TestQuantized.test_small_matrix",