Skip to content

Commit 378bb38

Browse files
committed
Skip some tests on TF GPU CI
1 parent 5a333ec commit 378bb38

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

Diff for: keras/src/layers/core/dense_test.py

+3
Original file line numberDiff line numberDiff line change
@@ -482,6 +482,7 @@ def test_quantize_invalid_mode(self, mode):
482482
("float8", "float8_from_mixed_bfloat16", 8, 0),
483483
)
484484
@pytest.mark.requires_trainable_backend
485+
@pytest.mark.skipif(testing.tensorflow_uses_gpu(), reason="Segfault")
485486
def test_quantize_dtype_argument(
486487
self, dtype, num_trainable_weights, num_non_trainable_weights
487488
):
@@ -498,6 +499,7 @@ def test_quantize_dtype_argument(
498499
)
499500

500501
@pytest.mark.requires_trainable_backend
502+
@pytest.mark.skipif(testing.tensorflow_uses_gpu(), reason="Segfault")
501503
def test_quantize_int8_when_lora_enabled(self):
502504
# Note that saving and loading with lora_enabled and quantized are
503505
# 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):
579581
)
580582

581583
@pytest.mark.requires_trainable_backend
584+
@pytest.mark.skipif(testing.tensorflow_uses_gpu(), reason="Segfault")
582585
def test_quantize_float8(self):
583586
import ml_dtypes
584587

0 commit comments

Comments
 (0)