Skip to content

GitLab

Commit be8c2556 authored by Allen Wang's avatar Allen Wang Committed by A. Unique TensorFlower
Browse files

Add in MLIR benchmarking for ResNet. 

PiperOrigin-RevId: 321039601
parent 5df3d2e3
Hide whitespace changes
Inline Side-by-side
Showing with 9 additions and 0 deletions
official/benchmark/resnet_ctl_imagenet_benchmark.py
View file @ be8c2556
...@@ -389,6 +389,15 @@ class Resnet50CtlBenchmarkBase(CtlBenchmark): ...@@ -389,6 +389,15 @@ class Resnet50CtlBenchmarkBase(CtlBenchmark):
FLAGS.dtype = 'bf16' FLAGS.dtype = 'bf16'
self._run_and_report_benchmark() self._run_and_report_benchmark()
@owner_utils.Owner('tf-graph-compiler')
def benchmark_2x2_tpu_bf16_mlir(self):
self._setup()
self._set_df_common()
FLAGS.batch_size = 1024
FLAGS.dtype = 'bf16'
tf.config.experimental.enable_mlir_bridge()
self._run_and_report_benchmark()
def benchmark_4x4_tpu_bf16(self): def benchmark_4x4_tpu_bf16(self):
self._setup() self._setup()
self._set_df_common() self._set_df_common()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment