Skip to content

Commit 2aa84ae

Browse files
Disable MLIR bridge for the tests that MLIR bridge silently fails
PiperOrigin-RevId: 679257946
1 parent ab9494c commit 2aa84ae

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tf_keras/distribute/ctl_correctness_test.py

+2
Original file line numberDiff line numberDiff line change
@@ -295,6 +295,8 @@ def test_dnn_correctness_minus_tpus(
295295
sync_batchnorm,
296296
jit_compile,
297297
):
298+
tf.config.experimental.disable_mlir_bridge()
299+
298300
# TODO(anjs): Identify why this particular V1 optimizer needs a higher
299301
# tol.
300302
if (

0 commit comments

Comments
 (0)