diff --git a/.github/workflows/integration-tests-nvidia.yml b/.github/workflows/integration-tests-nvidia.yml index e6115b7953d3..1c5ac7f0fd87 100644 --- a/.github/workflows/integration-tests-nvidia.yml +++ b/.github/workflows/integration-tests-nvidia.yml @@ -69,19 +69,12 @@ jobs: - name: Update PATH run: | echo "$HOME/.local/bin" >> $GITHUB_PATH - - name: Setup Python environment for GB200 - if: ${{ startsWith(matrix.config.runner_type, 'nvidia-gb200') }} + - name: Setup Python environment for GB200 and H100 + if: ${{ startsWith(matrix.config.runner_type, 'nvidia-gb200') || startsWith(matrix.config.runner_type, 'nvidia-h100') }} run: | echo "/venv/bin" >> $GITHUB_PATH echo "VIRTUAL_ENV=/venv" >> $GITHUB_ENV echo "PYTHONHOME=" >> $GITHUB_ENV - - name: Setup Python environment for H100 - if: ${{ startsWith(matrix.config.runner_type, 'nvidia-h100') }} - run: | - sudo chown -R "$(id -u):$(id -g)" /venv - echo "/venv/bin" >> $GITHUB_PATH - echo "VIRTUAL_ENV=/venv" >> $GITHUB_ENV - echo "PYTHONHOME=" >> $GITHUB_ENV - name: Install Triton env: CUDA_HOME: "/usr/local/cuda"