#8694: add --collect-only for sweep tests

tests/ttnn/sweep_tests/run_sweeps.py

@@ -7,7 +7,7 @@
 import ttnn
 
 
-from sweeps import run_sweeps, print_report
+from sweeps import run_sweeps, print_report, collect_only
 
 
 def convert_string_to_list(string):
@@ -21,16 +21,22 @@ def convert_string_to_list(string):
 
 def main():
     parser = argparse.ArgumentParser()
-    parser.add_argument("--include", type=str)
-
-    include = parser.parse_args().include
-
-    include = convert_string_to_list(include)
-
-    device = ttnn.open_device(device_id=0)
-    table_names = run_sweeps(device=device, include=include)
-    ttnn.close_device(device)
-    print_report(table_names=table_names)
+    parser.add_argument(
+        "--include", type=str, help="Comma separated list of sweep names to include eg) --include sweep1.py,sweep2.py"
+    )
+    parser.add_argument(
+        "--collect-only", action="store_true", help="Print the sweeps that will be run but do not run them"
+    )
+
+    args = parser.parse_args()
+    include = convert_string_to_list(args.include)
+    if args.collect_only:
+        collect_only(include=include)
+    else:
+        device = ttnn.open_device(device_id=0)
+        table_names = run_sweeps(device=device, include=include)
+        ttnn.close_device(device)
+        print_report(table_names=table_names)
 
 
 if __name__ == "__main__":

tests/ttnn/sweep_tests/sweeps/__init__.py

@@ -171,6 +171,14 @@ def name_to_string(name):
     return table_name
 
 
+def collect_only(*, include):
+    for file_name in sorted(SWEEP_SOURCES_DIR.glob("**/*.py")):
+        sweep_name = get_sweep_name(file_name)
+        if include and sweep_name not in include:
+            continue
+        print(sweep_name)
+
+
 def run_sweeps(*, device, include):
     table_names = []
     for file_name in sorted(SWEEP_SOURCES_DIR.glob("**/*.py")):