Disable MLIR bridge for the tests that MLIR bridge silently fails

PiperOrigin-RevId: 679257946
keras-team · Sep 26, 2024 · 2aa84ae · 2aa84ae
1 parent ab9494c
commit 2aa84ae
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 (