adding tests

Author: cquil11

.github/workflows/e2e-tests.yml

@@ -1,5 +1,5 @@
 name: End-to-End Tests
-run-name: e2e Test - ${{ github.event.inputs.generate-cli-command }}
+run-name: e2e Test - ${{ inputs.test-name || github.event.inputs.generate-cli-command }}
 
 on:
     workflow_dispatch:
@@ -8,6 +8,10 @@ on:
                 description: "Command passed to generate matrix script"
                 required: true
                 type: string
+            test-name:
+                description: "Name for this test run"
+                required: false
+                type: string
 
 jobs:
     get-jobs:

utils/matrix-logic/generate_sweep_configs.py

@@ -260,6 +260,12 @@ def generate_runner_model_sweep_config(args, all_config_data, runner_data):
 
         is_multinode = val.get(Fields.MULTINODE.value, False)
 
+        # Skip configs that don't match the requested node type
+        if args.single_node and is_multinode:
+            continue
+        if args.multi_node and not is_multinode:
+            continue
+
         # Get model code for exp_name
         model_code = val[Fields.MODEL_PREFIX.value]
         # Get disagg value, defaulting to False if not specified
@@ -517,6 +523,18 @@ def main():
         required=False,
         help='Filter runner nodes by substring match (e.g., "mi300x-amd" to only include nodes containing that string)'
     )
+    test_node_group = test_config_parser.add_mutually_exclusive_group(
+        required=True)
+    test_node_group.add_argument(
+        '--single-node',
+        action='store_true',
+        help='Generate single-node configurations only'
+    )
+    test_node_group.add_argument(
+        '--multi-node',
+        action='store_true',
+        help='Generate multi-node configurations only'
+    )
     test_config_parser.add_argument(
         '-h', '--help',
         action='help',

utils/matrix-logic/test_generate_sweep_configs.py

@@ -484,11 +484,13 @@ class TestGenerateRunnerModelSweepConfig:
 
     @pytest.fixture
     def runner_sweep_args(self):
-        """Args for runner-model-sweep command."""
+        """Args for runner-model-sweep command (single-node)."""
         args = argparse.Namespace()
         args.runner_type = "mi300x"
         args.runner_config = "runners.yaml"
         args.runner_node_filter = None
+        args.single_node = True
+        args.multi_node = False
         return args
 
     def test_basic_runner_sweep(self, sample_single_node_config, sample_runner_config, runner_sweep_args):