use 15 cores; build for all rocm versions

facebookresearch · Sep 3, 2024 · 146a97c · 146a97c
1 parent ff23584
commit 146a97c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/.github/workflows/wheels.yml b/.github/workflows/wheels.yml
@@ -40,7 +40,7 @@ jobs:
           for cu in CU_VERSIONS[1:]:
             PY_CU.append((PY_VERSIONS[-1], cu))
           print("Limited matrix PY_CU", PY_CU)
-          PY_ROCM = [(PY_VERSIONS[-1], ROCM_VERSIONS[-1])]
+          PY_ROCM = [(PY_VERSIONS[-1], rocm_v) for rocm_v in ROCM_VERSIONS]
 
         include = []
         for os in ['8-core-ubuntu', 'windows-8-core']:

diff --git 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=12" >> ${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