zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-12-16 01:49:05 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Disable the test for CUDA

Browse Source
This commit is contained in:
Angelos Katharopoulos
2025-07-04 19:17:45 -07:00
parent 2d0f452aae
commit b28577289e
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
python/tests/cuda_skip.py
View File

@@ -8,6 +8,9 @@ cuda_skip = {
# Gather matmul NYI # Gather matmul NYI
"TestBlas.test_gather_matmul", "TestBlas.test_gather_matmul",
"TestBlas.test_gather_matmul_grad", "TestBlas.test_gather_matmul_grad",
"TestBlas.test_gather_mm_sorted",
# Segmented matmul NYI
"TestBlas.test_segmented_mm",
# Scan NYI # Scan NYI
"TestArray.test_api", "TestArray.test_api",
"TestAutograd.test_cumprod_grad", "TestAutograd.test_cumprod_grad",