From 3ad80ca351936181ff16a30e932ea1268c001447 Mon Sep 17 00:00:00 2001 From: Anna R Date: Fri, 30 Oct 2020 16:05:54 -0700 Subject: [PATCH] Fix and re-enable cwise ops benchmark. PiperOrigin-RevId: 339956677 Change-Id: Id4cea5f63b3f257170c121d74e9cb2995a10b629 --- tensorflow/core/kernels/cwise_ops_test.cc | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tensorflow/core/kernels/cwise_ops_test.cc b/tensorflow/core/kernels/cwise_ops_test.cc index f0a0d7bccbd..253974ba688 100644 --- a/tensorflow/core/kernels/cwise_ops_test.cc +++ b/tensorflow/core/kernels/cwise_ops_test.cc @@ -184,7 +184,8 @@ Graph* CubeWithMulSquare(int num) { void BM_##DEVICE##_Cube_##Impl(::testing::benchmark::State& state) { \ const int num = state.range(0); \ \ - test::Benchmark(#DEVICE, Impl(num)).Run(state.iterations()); \ + test::Benchmark(#DEVICE, Impl(num), /*old_benchmark_api*/ false) \ + .Run(state); \ const int64 tot = static_cast(state.iterations()) * num; \ state.SetItemsProcessed(tot); \ state.SetBytesProcessed(tot * sizeof(float)); \