From b28577289e9795d25de7c552f419c8a119f6c500 Mon Sep 17 00:00:00 2001
From: Angelos Katharopoulos <a_katharopoulos@apple.com>
Date: Fri, 4 Jul 2025 19:17:45 -0700
Subject: [PATCH] Disable the test for CUDA

---
 python/tests/cuda_skip.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/python/tests/cuda_skip.py b/python/tests/cuda_skip.py
index fce92bacb..faca4a66f 100644
--- a/python/tests/cuda_skip.py
+++ b/python/tests/cuda_skip.py
@@ -8,6 +8,9 @@ cuda_skip = {
     # Gather matmul NYI
     "TestBlas.test_gather_matmul",
     "TestBlas.test_gather_matmul_grad",
+    "TestBlas.test_gather_mm_sorted",
+    # Segmented matmul NYI
+    "TestBlas.test_segmented_mm",
     # Scan NYI
     "TestArray.test_api",
     "TestAutograd.test_cumprod_grad",