From 0ddda2d65ebb85add1d6684b2f3892b131ffd354 Mon Sep 17 00:00:00 2001
From: Max Podkorytov <4273004+tenpercent@users.noreply.github.com>
Date: Fri, 6 Sep 2024 01:23:37 -0700
Subject: [PATCH] [ROCm] tweak rocm wheel build workflow to avoid ci timeouts
 (#1091)

* try increasing number of workers to 12 to avoid ci timeouts

* try 16-core runner for rocm

* use 15 cores; build for all rocm versions

* revert many-rocm wheel build when getting CI signal; add comment about runner
---
 .github/workflows/wheels.yml       | 2 +-
 .github/workflows/wheels_build.yml | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

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