diff --git a/.github/workflows/triton-test.yaml b/.github/workflows/triton-test.yaml index 191d445932..3482ea0147 100644 --- a/.github/workflows/triton-test.yaml +++ b/.github/workflows/triton-test.yaml @@ -60,7 +60,7 @@ jobs: build-triton: if: ${{ (!github.event.pull_request || github.event.pull_request.draft == false) && (github.event_name != 'pull_request' || github.event.action != 'labeled' || github.event.label.name == 'ci:triton-300x') }} name: Build Triton Wheel - runs-on: build-only-aiter + runs-on: linux-aiter-mi35x-1 needs: [check-signal] env: DOCKER_IMAGE: "rocm/pytorch:latest"