diff --git a/.github/workflows/nvidia_workflow.yml b/.github/workflows/nvidia_workflow.yml index a680e3e4..544c9de9 100644 --- a/.github/workflows/nvidia_workflow.yml +++ b/.github/workflows/nvidia_workflow.yml @@ -18,23 +18,8 @@ on: run-name: 'NVIDIA Job - ${{ github.event.inputs.run_id }}' jobs: - select-runner: - runs-on: ubuntu-latest - outputs: - runner: ${{ steps.pick.outputs.runner }} - steps: - - id: pick - run: | - runners=("b200-02-gpu0" "b200-02-gpu1" "b200-02-gpu2" "b200-02-gpu3" "b200-02-gpu4" "b200-02-gpu5" "b200-02-gpu6" "b200-02-gpu7") - index=$(( ${{ github.run_number }} % 8 )) - echo "runner=${runners[$index]}" >> $GITHUB_OUTPUT - run: - needs: select-runner - runs-on: ${{ needs.select-runner.outputs.runner }} - concurrency: - group: ${{ needs.select-runner.outputs.runner }} - cancel-in-progress: false + runs-on: [self-hosted, nvidia-docker-b200-8-x86-64] timeout-minutes: 10 steps: - uses: actions/checkout@v3