diff --git a/.github/workflows/pr-test.yml b/.github/workflows/pr-test.yml index 7e196f01685..93aedcac2bf 100644 --- a/.github/workflows/pr-test.yml +++ b/.github/workflows/pr-test.yml @@ -25,7 +25,7 @@ jobs: with: gpu_count: 0 cpu_runner: true - container_image: "radixark/miles:dev-submodule-test" # TEMP — REMOVE BEFORE MERGE + container_image: "radixark/miles:dev-phase1-test" # TEMP — REMOVE BEFORE MERGE execute_command: "python tests/ci/run_suite.py --hw cpu --suite stage-a-fast" secrets: inherit @@ -37,7 +37,7 @@ jobs: uses: ./.github/workflows/_run-ci.yml with: gpu_count: 8 - container_image: "radixark/miles:dev-submodule-test" # TEMP — REMOVE BEFORE MERGE + container_image: "radixark/miles:dev-phase1-test" # TEMP — REMOVE BEFORE MERGE execute_command: "pytest tests/e2e/fsdp/test_qwen3_4B_fsdp_true_on_policy.py -x -v" secrets: inherit @@ -52,7 +52,7 @@ jobs: uses: ./.github/workflows/_run-ci.yml with: gpu_count: 1 - container_image: "radixark/miles:dev-submodule-test" # TEMP — REMOVE BEFORE MERGE + container_image: "radixark/miles:dev-phase1-test" # TEMP — REMOVE BEFORE MERGE execute_command: "python tests/ci/run_suite.py --hw cuda --suite stage-b-fast-1-gpu" secrets: inherit @@ -65,7 +65,7 @@ jobs: uses: ./.github/workflows/_run-ci.yml with: gpu_count: 8 - container_image: "radixark/miles:dev-submodule-test" # TEMP — REMOVE BEFORE MERGE + container_image: "radixark/miles:dev-phase1-test" # TEMP — REMOVE BEFORE MERGE execute_command: "python tests/ci/run_suite.py --hw cuda --suite stage-b-sglang-8-gpu" secrets: inherit @@ -78,7 +78,7 @@ jobs: uses: ./.github/workflows/_run-ci.yml with: gpu_count: 8 - container_image: "radixark/miles:dev-submodule-test" # TEMP — REMOVE BEFORE MERGE + container_image: "radixark/miles:dev-phase1-test" # TEMP — REMOVE BEFORE MERGE execute_command: "python tests/ci/run_suite.py --hw cuda --suite stage-b-short-8-gpu" secrets: inherit @@ -91,7 +91,7 @@ jobs: uses: ./.github/workflows/_run-ci.yml with: gpu_count: 8 - container_image: "radixark/miles:dev-submodule-test" # TEMP — REMOVE BEFORE MERGE + container_image: "radixark/miles:dev-phase1-test" # TEMP — REMOVE BEFORE MERGE execute_command: "python tests/ci/run_suite.py --hw cuda --suite stage-c-fsdp-8-gpu" secrets: inherit @@ -108,7 +108,7 @@ jobs: uses: ./.github/workflows/_run-ci.yml with: gpu_count: 8 - container_image: "radixark/miles:dev-submodule-test" # TEMP — REMOVE BEFORE MERGE + container_image: "radixark/miles:dev-phase1-test" # TEMP — REMOVE BEFORE MERGE execute_command: "python tests/ci/run_suite.py --hw cuda --suite stage-c-megatron-8-gpu --auto-partition-id ${{ matrix.partition_id }} --auto-partition-size 3" secrets: inherit @@ -121,7 +121,7 @@ jobs: uses: ./.github/workflows/_run-ci.yml with: gpu_count: 8 - container_image: "radixark/miles:dev-submodule-test" # TEMP — REMOVE BEFORE MERGE + container_image: "radixark/miles:dev-phase1-test" # TEMP — REMOVE BEFORE MERGE execute_command: "python tests/ci/run_suite.py --hw cuda --suite stage-c-precision-8-gpu" secrets: inherit @@ -134,7 +134,7 @@ jobs: uses: ./.github/workflows/_run-ci.yml with: gpu_count: 8 - container_image: "radixark/miles:dev-submodule-test" # TEMP — REMOVE BEFORE MERGE + container_image: "radixark/miles:dev-phase1-test" # TEMP — REMOVE BEFORE MERGE execute_command: "python tests/ci/run_suite.py --hw cuda --suite stage-c-ckpt-8-gpu" secrets: inherit @@ -147,7 +147,7 @@ jobs: uses: ./.github/workflows/_run-ci.yml with: gpu_count: 8 - container_image: "radixark/miles:dev-submodule-test" # TEMP — REMOVE BEFORE MERGE + container_image: "radixark/miles:dev-phase1-test" # TEMP — REMOVE BEFORE MERGE execute_command: "python tests/ci/run_suite.py --hw cuda --suite stage-c-long-8-gpu" secrets: inherit @@ -160,7 +160,7 @@ jobs: uses: ./.github/workflows/_run-ci.yml with: gpu_count: 8 - container_image: "radixark/miles:dev-submodule-test" # TEMP — REMOVE BEFORE MERGE + container_image: "radixark/miles:dev-phase1-test" # TEMP — REMOVE BEFORE MERGE execute_command: "python tests/ci/run_suite.py --hw cuda --suite stage-c-lora-8-gpu" secrets: inherit @@ -197,6 +197,6 @@ jobs: uses: ./.github/workflows/_run-ci.yml with: gpu_count: 8 - container_image: "radixark/miles:dev-submodule-test" # TEMP — REMOVE BEFORE MERGE + container_image: "radixark/miles:dev-phase1-test" # TEMP — REMOVE BEFORE MERGE execute_command: "python tests/ci/run_suite.py --hw cuda --suite ${{ matrix.suite }}" secrets: inherit