No public description

PiperOrigin-RevId: 666412764

Author: tensorflower-gardener

official/benchmark/bert_pretrain_benchmark.py

@@ -228,7 +228,6 @@ def benchmark_perf_2x2_tpu_bf16_seq128_10k_steps_mlir(self):
         'benchmark_perf_2x2_tpu_bf16_seq128_10k_steps_mlir')
     summary_path = os.path.join(FLAGS.model_dir,
                                 'summaries/training_summary.txt')
-    tf.config.experimental.enable_mlir_bridge()
     # Disable accuracy check.
     self._run_and_report_benchmark(
         summary_path=summary_path,
@@ -267,7 +266,6 @@ def benchmark_perf_4x4_tpu_bf16_seq128_10k_steps_mlir(self):
         'benchmark_perf_4x4_tpu_bf16_seq128_10k_steps_mlir')
     summary_path = os.path.join(FLAGS.model_dir,
                                 'summaries/training_summary.txt')
-    tf.config.experimental.enable_mlir_bridge()
     # Disable accuracy check.
     self._run_and_report_benchmark(
         summary_path=summary_path,

official/benchmark/ncf_keras_benchmark.py

@@ -414,7 +414,6 @@ def benchmark_2x2_tpu_mlir(self):
     FLAGS.keras_use_ctl = True
     FLAGS.num_gpus = 0
     FLAGS.train_epochs = 1
-    tf.config.experimental.enable_mlir_bridge()
     self._run_and_report_benchmark()
 
 

official/benchmark/resnet_ctl_imagenet_benchmark.py

@@ -403,7 +403,6 @@ def benchmark_2x2_tpu_bf16_mlir(self):
     self._set_df_common()
     FLAGS.batch_size = 1024
     FLAGS.dtype = 'bf16'
-    tf.config.experimental.enable_mlir_bridge()
     FLAGS.model_dir = self._get_model_dir('benchmark_2x2_tpu_bf16_mlir')
     self._run_and_report_benchmark()
 
@@ -426,7 +425,6 @@ def benchmark_4x4_tpu_bf16_mlir(self):
     FLAGS.batch_size = 4096
     FLAGS.dtype = 'bf16'
     FLAGS.model_dir = self._get_model_dir('benchmark_4x4_tpu_bf16_mlir')
-    tf.config.experimental.enable_mlir_bridge()
     self._run_and_report_benchmark()
 
   def benchmark_8x8_tpu_bf16(self):
@@ -444,7 +442,6 @@ def benchmark_8x8_tpu_bf16_mlir(self):
     FLAGS.batch_size = 8192
     FLAGS.dtype = 'bf16'
     FLAGS.model_dir = self._get_model_dir('benchmark_8x8_tpu_bf16_mlir')
-    tf.config.experimental.enable_mlir_bridge()
     self._run_and_report_benchmark()
 
   def benchmark_8x8_tpu(self):
@@ -460,7 +457,6 @@ def benchmark_8x8_tpu_mlir(self):
     self._set_df_common()
     FLAGS.batch_size = 8192
     FLAGS.model_dir = self._get_model_dir('benchmark_8x8_tpu_mlir')
-    tf.config.experimental.enable_mlir_bridge()
     self._run_and_report_benchmark()
 
   def benchmark_8x16_tpu_bf16(self):

official/benchmark/retinanet_benchmark.py

@@ -299,7 +299,6 @@ def benchmark_2x2_tpu_coco_mlir(self):
     params['train']['iterations_per_loop'] = 500
     FLAGS.model_dir = self._get_model_dir('real_benchmark_2x2_tpu_coco_mlir')
     FLAGS.strategy_type = 'tpu'
-    tf.config.experimental.enable_mlir_bridge()
     self._run_and_report_benchmark(params, do_eval=False, warmup=0)
 
   @flagsaver.flagsaver
@@ -312,7 +311,6 @@ def benchmark_4x4_tpu_coco_mlir(self):
     params['train']['iterations_per_loop'] = 500
     FLAGS.model_dir = self._get_model_dir('real_benchmark_4x4_tpu_coco_mlir')
     FLAGS.strategy_type = 'tpu'
-    tf.config.experimental.enable_mlir_bridge()
     self._run_and_report_benchmark(params, do_eval=False, warmup=0)
 
   @flagsaver.flagsaver