Skip sparse tests on GPU

keras-team · Jun 25, 2024 · c03e7b0 · c03e7b0
1 parent 5d78ec2
commit c03e7b0
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/keras/src/ops/numpy_test.py b/keras/src/ops/numpy_test.py
@@ -433,6 +433,7 @@ def test_matmul(self):
             y = KerasTensor((2, 3, 4))
             knp.matmul(x, y)
 
+    @pytest.mark.skipif(testing.tensorflow_uses_gpu(), reason="Segfault")
     def test_matmul_sparse(self):
         x = KerasTensor((2, 3), sparse=True)
         y = KerasTensor((3, 2))