From 9afd6f27b1eea96a7b137225f7ba67f0c8738e2a Mon Sep 17 00:00:00 2001
From: "A. Unique TensorFlower" <gardener@tensorflow.org>
Date: Thu, 19 Sep 2024 14:18:40 -0700
Subject: [PATCH] Disable MLIR bridge for the tests that MLIR bridge silently
 fails

PiperOrigin-RevId: 676560954
---
 tf_keras/distribute/ctl_correctness_test.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/tf_keras/distribute/ctl_correctness_test.py b/tf_keras/distribute/ctl_correctness_test.py
index 013b68324..a77681254 100644
--- 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 (