Disable MLIR bridge for the tests that MLIR bridge silently fails

PiperOrigin-RevId: 676560954
keras-team · Sep 26, 2024 · 9afd6f2 · 9afd6f2
1 parent ab9494c
commit 9afd6f2
Showing 1 changed file with 3 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,9 @@ 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 (