diff --git a/.github/workflows/wheels.yml b/.github/workflows/wheels.yml index e23a3d6186..bbd888dd33 100644 --- a/.github/workflows/wheels.yml +++ b/.github/workflows/wheels.yml @@ -60,7 +60,7 @@ jobs: if os == 'windows-8-core': continue include.append(dict( - os=os, + os="16-core-ubuntu", # use for ROCm wheels only to avoid CI timeouts python=python, torch_version=torch_version, toolkit_type="rocm", diff --git a/.github/workflows/wheels_build.yml b/.github/workflows/wheels_build.yml index 36ad288bea..47c81d5aaa 100644 --- a/.github/workflows/wheels_build.yml +++ b/.github/workflows/wheels_build.yml @@ -153,7 +153,7 @@ jobs: amdgpu-install -y --usecase=rocm --no-dkms echo "ROCM_PATH=/opt/rocm" >> ${GITHUB_ENV} echo "PATH=$PATH:/opt/rocm/bin" >> ${GITHUB_ENV} - echo "MAX_JOBS=7" >> ${GITHUB_ENV} + echo "MAX_JOBS=15" >> ${GITHUB_ENV} - name: Install dependencies run: $PY -m pip install wheel setuptools twine -r requirements.txt --extra-index-url $PYTORCH_INDEX_URL