ci: expose cpu_parallel_mode and run planner tests with -n 0 (#8378)

Signed-off-by: Dmitry Tokarev <dtokarev@nvidia.com>
Co-authored-by: Claude Opus 4.7 (1M context) <noreply@anthropic.com>

.github/workflows/build-test-distribute-flavor-matrix.yml

@@ -42,6 +42,11 @@ on:
         required: false
         type: number
         default: 10
+      cpu_parallel_mode:
+        description: 'pytest-xdist mode for CPU-only tests: auto, none/0, or a worker count'
+        required: false
+        type: string
+        default: 'auto'
       run_single_gpu_tests:
         description: 'Whether to run single GPU tests'
         required: false
@@ -164,6 +169,7 @@ jobs:
       run_cpu_only_tests: ${{ inputs.run_cpu_only_tests }}
       cpu_only_test_markers: ${{ inputs.cpu_only_test_markers }}
       cpu_only_test_timeout_minutes: ${{ inputs.cpu_only_test_timeout_minutes }}
+      cpu_parallel_mode: ${{ inputs.cpu_parallel_mode }}
       run_single_gpu_tests: ${{ inputs.run_single_gpu_tests }}
       single_gpu_test_markers: ${{ inputs.single_gpu_test_markers }}
       single_gpu_test_timeout_minutes: ${{ inputs.single_gpu_test_timeout_minutes }}

.github/workflows/build-test-distribute-flavor.yml

@@ -52,6 +52,11 @@ on:
         required: false
         type: number
         default: 10
+      cpu_parallel_mode:
+        description: 'pytest-xdist mode for CPU-only tests: auto, none/0, or a worker count'
+        required: false
+        type: string
+        default: 'auto'
       run_single_gpu_tests:
         description: 'Whether to run single GPU tests'
         required: false
@@ -290,7 +295,7 @@ jobs:
           test_type: "pre_merge_cpu"
           platform_arch: ${{ matrix.arch }}
           hf_token: ${{ secrets.HF_TOKEN }}
-          parallel_mode: 'auto'
+          parallel_mode: ${{ inputs.cpu_parallel_mode }}
           dind_as_sidecar: 'true'
 
       # Run GPU tests sequentially (only on amd64 runners with GPU)

.github/workflows/post-merge-ci.yml

@@ -36,6 +36,7 @@ jobs:
       run_cpu_only_tests: true
       cpu_only_test_markers: '(pre_merge or post_merge) and planner and gpu_0'
       cpu_only_test_timeout_minutes: 30
+      cpu_parallel_mode: '0'
       run_single_gpu_tests: false
       run_multi_gpu_tests: false
       copy_to_acr: false

.github/workflows/pr.yaml

@@ -362,11 +362,12 @@ jobs:
       amd_runner: prod-tester-amd-gpu-v1 # TODO: CPU only DinD runner for dynamo repo
       target_tag_plain: ${{ needs.planner-build.outputs.target_tag_plain }}
       cuda_version: '[""]'
-      platform: '["amd64"]'
+      platform: '["amd64", "arm64"]'
       run_sanity_check: false
       run_cpu_only_tests: true
       cpu_only_test_markers: 'pre_merge and planner and gpu_0'
       cpu_only_test_timeout_minutes: 30
+      cpu_parallel_mode: '0'
       run_gpu_tests: false
     secrets: inherit
 

.github/workflows/shared-test.yml

@@ -49,6 +49,11 @@ on:
         required: false
         type: number
         default: 10
+      cpu_parallel_mode:
+        description: 'pytest-xdist mode for CPU-only tests: auto, none/0, or a worker count'
+        required: false
+        type: string
+        default: 'auto'
       run_gpu_tests:
         description: 'Whether to run GPU tests'
         required: false
@@ -155,7 +160,7 @@ jobs:
           test_type: "pre_merge_cpu"
           platform_arch: ${{ matrix.platform }}
           hf_token: ${{ secrets.HF_TOKEN }}
-          parallel_mode: 'auto'
+          parallel_mode: ${{ inputs.cpu_parallel_mode }}
           dind_as_sidecar: 'true'
       - name: Run GPU tests (sequential)
         timeout-minutes: ${{ inputs.gpu_test_timeout_minutes }}