From 9cfe208f8a4e675674e181e3cbe7c0d47b078791 Mon Sep 17 00:00:00 2001 From: Francois Chollet Date: Tue, 25 Jun 2024 10:19:41 -0700 Subject: [PATCH] Skip TF sparse test on GPU --- keras/src/ops/numpy_test.py | 1 + 1 file changed, 1 insertion(+) diff --git a/keras/src/ops/numpy_test.py b/keras/src/ops/numpy_test.py index 76952cef321..2bfe4a6378e 100644 --- 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