experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix remote benchmarks with explicit execution_mode.

Browse Source 
PiperOrigin-RevId: 291986433
Change-Id: I471538b81b763e5e2ea0eaa88af4eca5cb37251d
This commit is contained in:
Haoyu Zhang 2020-01-28 12:08:50 -08:00 committed by TensorFlower Gardener
parent 73675deef7
commit 4a9a77dd2f
1 changed files with 1 additions and 1 deletions

2
tensorflow/python/eager/remote_benchmarks_test.py
View File

@ -84,7 +84,7 @@ class RemoteWorkerMicroBenchmarks(test.Benchmark):
self._cached_server2 = server_lib.Server.create_local_server()
self._cached_server_target2 = self._cached_server2.target[len("grpc://"):]
def _run(self, func, num_iters=1000, execution_mode=None):
def _run(self, func, num_iters=1000, execution_mode=context.ASYNC):
total_time = run_benchmark(func, num_iters, execution_mode)
mean_us = total_time * 1e6 / num_iters
self.report_benchmark(