Merge pull request #46060 from jackd:bm-main-argv

PiperOrigin-RevId: 356369458 Change-Id: Ie89c8552a44dc6c42802ba9ecaf21bded401708c
2021-02-08 15:45:09 -08:00 · 2021-02-08 15:45:09 -08:00 · 21ec8483b2
commit 21ec8483b2
parent 885c55ca95 461712aa35
1 changed files with 2 additions and 1 deletions
--- a/tensorflow/python/platform/googletest.py
+++ b/tensorflow/python/platform/googletest.py
@ -63,7 +63,8 @@ def main(argv=None):  # pylint: disable=function-redefined
    if args is None:
      args = sys.argv
    return app.run(main=g_main, argv=args)
-  benchmark.benchmarks_main(true_main=main_wrapper)
+
+  benchmark.benchmarks_main(true_main=main_wrapper, argv=argv)


 def GetTempDir():