diff --git a/tensorflow/python/platform/googletest.py b/tensorflow/python/platform/googletest.py index cc3194be887..a88344f5185 100644 --- 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():