diff --git a/.github/workflows/aiter-test.yaml b/.github/workflows/aiter-test.yaml index a650c5a1f2..a122cf83c4 100644 --- a/.github/workflows/aiter-test.yaml +++ b/.github/workflows/aiter-test.yaml @@ -42,16 +42,16 @@ jobs: pr_title="${{ github.event.pull_request.title }}" if [[ "${{ github.ref }}" == "refs/heads/main" ]]; then echo "It's main branch, running tests on MI325 and MI35X..." - echo 'standard_runners=["aiter-1gpu-runner","aiter-mi355-1gpu"]' >> "$GITHUB_OUTPUT" - echo 'multigpu_runners=["aiter-8gpu-runner","aiter-mi355-8gpu"]' >> "$GITHUB_OUTPUT" + echo 'standard_runners=["aiter-mi355-1gpu"]' >> "$GITHUB_OUTPUT" + echo 'multigpu_runners=["aiter-mi355-8gpu"]' >> "$GITHUB_OUTPUT" elif echo "$pr_title" | grep -qi "mi35x"; then echo "PR title contains 'MI35X', running tests on MI325 and MI35X..." - echo 'standard_runners=["aiter-1gpu-runner"]' >> "$GITHUB_OUTPUT" - echo 'multigpu_runners=["aiter-8gpu-runner"]' >> "$GITHUB_OUTPUT" + echo 'standard_runners=["aiter-mi355-1gpu"]' >> "$GITHUB_OUTPUT" + echo 'multigpu_runners=["aiter-mi355-8gpu"]' >> "$GITHUB_OUTPUT" else echo "Not main branch and PR title does not contain mi35x, only running on MI325..." - echo 'standard_runners=["aiter-1gpu-runner"]' >> "$GITHUB_OUTPUT" - echo 'multigpu_runners=["aiter-8gpu-runner"]' >> "$GITHUB_OUTPUT" + echo 'standard_runners=["aiter-mi355-1gpu"]' >> "$GITHUB_OUTPUT" + echo 'multigpu_runners=["aiter-mi355-8gpu"]' >> "$GITHUB_OUTPUT" fi echo "$GITHUB_OUTPUT" diff --git a/.github/workflows/sglang_downstream.yaml b/.github/workflows/sglang_downstream.yaml index 3c66e3b6ec..7e71240448 100644 --- a/.github/workflows/sglang_downstream.yaml +++ b/.github/workflows/sglang_downstream.yaml @@ -27,10 +27,10 @@ jobs: sglang: name: sglang integration needs: [check-signal] - runs-on: aiter-1gpu-runner + runs-on: aiter-mi355-1gpu env: SGL_BRANCH: v0.5.3 - GPU_ARCH: gfx942 + GPU_ARCH: gfx950 SGL_IMAGE: rocm/sgl-dev:v0.5.3.post3-rocm700-mi30x-20251019 GITHUB_REPO_URL: ${{ github.event.pull_request.head.repo.clone_url || 'https://github.com/ROCm/aiter.git' }} GITHUB_COMMIT_SHA: ${{ github.event.pull_request.head.sha || github.event.head_commit.id }} diff --git a/.github/workflows/vllm_benchmark.yaml b/.github/workflows/vllm_benchmark.yaml index d800d208f5..f5a9b46c8b 100644 --- a/.github/workflows/vllm_benchmark.yaml +++ b/.github/workflows/vllm_benchmark.yaml @@ -34,7 +34,7 @@ jobs: build_vllm_image: if: ${{ !github.event.pull_request.head.repo.fork }} needs: [check-signal] - runs-on: aiter-1gpu-runner + runs-on: aiter-k8s-build steps: - name: Checkout aiter repo