Add run_config

pytorch · May 10, 2024 · 65aac18 · 65aac18
1 parent 52bdbaa
commit 65aac18
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/userbenchmark/group_bench/run_config.py b/userbenchmark/group_bench/run_config.py
@@ -7,7 +7,7 @@
 def _get_models(models: Optional[List[str]]=None, model_set: Optional[List[str]]=None) -> List[str]:
     result = set(models) if models else set()
     for s in model_set:
-        result.add(list_extended_models(s))
+        result.union(set(list_extended_models(s)))
     return list(result) 
 
 def config_obj_to_model_configs(config: Dict[str, Any]) -> Dict[str, Dict[str, List[TorchBenchModelConfig]]]:

diff --git a/userbenchmark/torch-nightly/run.py b/userbenchmark/torch-nightly/run.py
@@ -25,7 +25,7 @@ def parse_args(args):
 def run(args: List[str]):
     args = parse_args(args)
     assert os.path.exists(args.config), f"Expect an existing benchmark config file, get path: {args.config}."
-    benchmark_result = get_output_json(BM_NAME, run_benchmark_group_config(config_file=args.config, dryrun=args.dryrun))
+    benchmark_result = get_output_json(BM_NAME, run_benchmark_group_config(group_config_file=args.config, dryrun=args.dryrun))
     benchmark_result["environ"]["benchmark_style"] = "group_bench"
     benchmark_result_json = json.dumps(benchmark_result, indent=4)
     with open(args.output, "w") as fp: