test

Author: cquil11

.github/workflows/run-sweep.yml

@@ -83,17 +83,17 @@ jobs:
     #         decode-dp-attn: ${{ matrix.config.decode.dp-attn }}
     #         decode-additional-settings: ${{ toJson(matrix.config.decode.additional-settings) }}
 
-    test-sweep-single-node:
+    sweep-single-node-1k1k:
         needs: get-jobs
-        if: ${{ needs.get-jobs.outputs.single-node-search-space-config != '[]' }}
+        if: ${{ needs.get-jobs.outputs.single-node-search-space-config.single_node['1k1k'] != '[]' }}
         uses: ./.github/workflows/benchmark-tmpl.yml
-        name: single-node /
+        name: single-node 1k1k /
         strategy:
             fail-fast: false
             matrix:
                 config: ${{ fromJson(needs.get-jobs.outputs.multi-node-search-space-config).single_node['1k1k'] }}
-        secrets: inherit
-        with:
+        secrets: inherit 
+        with: &single-node-inputs
             exp-name: ${{ matrix.config.exp-name }}
             isl: ${{ matrix.config.isl }}
             osl: ${{ matrix.config.osl }}
@@ -111,8 +111,32 @@ jobs:
             spec-decoding: ${{ matrix.config.spec-decoding }}
             disagg: ${{ matrix.config.disagg }}
 
+    sweep-single-node-1k8k:
+        needs: get-jobs
+        if: ${{ needs.get-jobs.outputs.single-node-search-space-config.single_node['1k8k'] != '[]' }}
+        uses: ./.github/workflows/benchmark-tmpl.yml
+        name: single-node 1k8k /
+        strategy:
+            fail-fast: false
+            matrix:
+                config: ${{ fromJson(needs.get-jobs.outputs.multi-node-search-space-config).single_node['1k8k'] }}
+        secrets: inherit 
+        with: *single-node-inputs
+
+    sweep-single-node-8k1k:
+        needs: get-jobs
+        if: ${{ needs.get-jobs.outputs.single-node-search-space-config.single_node['8k1k'] != '[]' }}
+        uses: ./.github/workflows/benchmark-tmpl.yml
+        name: single-node 8k1k /
+        strategy:
+            fail-fast: false
+            matrix:
+                config: ${{ fromJson(needs.get-jobs.outputs.multi-node-search-space-config).single_node['8k1k'] }}
+        secrets: inherit 
+        with: *single-node-inputs
+
     collect-results:
-        needs: [test-sweep-single-node]
+        needs: [sweep-single-node-1k1k, sweep-single-node-1k8k, sweep-single-node-8k1k]
         if: ${{ always() }}
         uses: ./.github/workflows/collect-results.yml
         secrets: inherit