Disable MLIR bridge for the tests that MLIR bridge silently fails

PiperOrigin-RevId: 676560954
keras-team · Sep 19, 2024 · bda0bad · bda0bad
1 parent c5f9773
commit bda0bad
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tf_keras/distribute/ctl_correctness_test.py b/tf_keras/distribute/ctl_correctness_test.py
@@ -295,6 +295,8 @@ def test_dnn_correctness_minus_tpus(
         sync_batchnorm,
         jit_compile,
     ):
+        tf.config.experimental.disable_mlir_bridge()
+
         # TODO(anjs): Identify why this particular V1 optimizer needs a higher
         # tol.
         if (