Add outdir arg

Change-Id: Ibc76668929262abc59afe5b601acf0dff7203bf2
Jeffwhen · Feb 14, 2023 · f89d3c8 · f89d3c8
1 parent 866ea0b
commit f89d3c8
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/python/tpu_perf/buildtree.py b/python/tpu_perf/buildtree.py
@@ -40,8 +40,10 @@ def __init__(self, root, args = None):
         self.root = root
         self.global_config = global_config = read_config(root) or dict()
         global_config['root'] = root
-        if 'outdir' not in global_config:
-            global_config['outdir'] = os.path.join(root, 'output')
+        outdir = 'output'
+        if args.outdir:
+            outdir = args.outdir
+        global_config['outdir'] = os.path.join(root, outdir)
 
         self.cases = []
         if not args.full:
@@ -73,6 +75,7 @@ def add_arguments(parser):
             'models', metavar='MODEL', type=str, nargs='*',
             help='model directories to run')
         parser.add_argument('--full', action='store_true', help='Run all cases')
+        parser.add_argument('--outdir', '-o', type=str, help='Output path')
         parser.add_argument('--list', '-l', type=str, help='Case list')
         parser.add_argument('--devices', '-d',
             type=int, nargs='*', help='Devices',