From 378bb38e9b5cf57f9f728c055c7d35641ada5e8f Mon Sep 17 00:00:00 2001 From: Francois Chollet Date: Mon, 12 Aug 2024 12:57:20 -0700 Subject: [PATCH] Skip some tests on TF GPU CI --- keras/src/layers/core/dense_test.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/keras/src/layers/core/dense_test.py b/keras/src/layers/core/dense_test.py index ef3ec8620c8..30ec6722613 100644 --- a/keras/src/layers/core/dense_test.py +++ b/keras/src/layers/core/dense_test.py @@ -482,6 +482,7 @@ def test_quantize_invalid_mode(self, mode): ("float8", "float8_from_mixed_bfloat16", 8, 0), ) @pytest.mark.requires_trainable_backend + @pytest.mark.skipif(testing.tensorflow_uses_gpu(), reason="Segfault") def test_quantize_dtype_argument( self, dtype, num_trainable_weights, num_non_trainable_weights ): @@ -498,6 +499,7 @@ def test_quantize_dtype_argument( ) @pytest.mark.requires_trainable_backend + @pytest.mark.skipif(testing.tensorflow_uses_gpu(), reason="Segfault") def test_quantize_int8_when_lora_enabled(self): # Note that saving and loading with lora_enabled and quantized are # lossy, so we use a weak correctness test for model outputs (atol=0.5). @@ -579,6 +581,7 @@ def test_quantize_int8_when_lora_enabled(self): ) @pytest.mark.requires_trainable_backend + @pytest.mark.skipif(testing.tensorflow_uses_gpu(), reason="Segfault") def test_quantize_float8(self): import ml_dtypes