Skip to content

Commit

Permalink
Skip some tests on TF GPU CI
Browse files Browse the repository at this point in the history
  • Loading branch information
fchollet committed Aug 12, 2024
1 parent 5a333ec commit 378bb38
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions keras/src/layers/core/dense_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
):
Expand All @@ -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).
Expand Down Expand Up @@ -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

Expand Down

0 comments on commit 378bb38

Please sign in to comment.