From 461712aa3526cf7209d8799e4c0b9a8f0f3f5a7e Mon Sep 17 00:00:00 2001
From: Dominic Jack <thedomjack@gmail.com>
Date: Wed, 30 Dec 2020 11:27:53 +1000
Subject: [PATCH] Propagated optional argv to benchmarks_main
---
tensorflow/python/platform/googletest.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/tensorflow/python/platform/googletest.py b/tensorflow/python/platform/googletest.py
index cc3194be887..1aa30c4f83c 100644
--- 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():