Propagated optional argv to benchmarks_main

2020-12-30 11:27:53 +10:00 · 2020-12-30 11:27:53 +10:00 · 461712aa35
commit 461712aa35
parent 7bdd911281
1 changed files with 1 additions and 1 deletions
--- a/tensorflow/python/platform/googletest.py
+++ b/tensorflow/python/platform/googletest.py
@ -63,7 +63,7 @@ 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():