Skip TF sparse test on GPU

keras-team · Jun 25, 2024 · 9cfe208 · 9cfe208
1 parent f0bae91
commit 9cfe208
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
@@ -2083,6 +2083,7 @@ def test_matmul(self):
         not backend.SUPPORTS_SPARSE_TENSORS,
         reason="Backend does not support sparse tensors.",
     )
+    @pytest.mark.skipif(testing.tensorflow_uses_gpu(), reason="Segfault")
     def test_matmul_sparse(self, dtype, x_shape, y_shape, x_sparse, y_sparse):
         if backend.backend() == "tensorflow":
             import tensorflow as tf