From 434f60bbce08bf98cb9c1924859383ad0acfc254 Mon Sep 17 00:00:00 2001
From: Jake Awe <jawe@nvidia.com>
Date: Tue, 14 May 2024 17:12:02 +0000
Subject: [PATCH] update manylinux ver

---
 .github/workflows/build-image.yaml |  1 +
 ci/compute-build-args.sh           | 15 ++++++++++-----
 2 files changed, 11 insertions(+), 5 deletions(-)

diff --git a/.github/workflows/build-image.yaml b/.github/workflows/build-image.yaml
index dc7e632..974816c 100644
--- a/.github/workflows/build-image.yaml
+++ b/.github/workflows/build-image.yaml
@@ -57,6 +57,7 @@ jobs:
           LINUX_VER: ${{ inputs.LINUX_VER }}
           PYTHON_VER: ${{ inputs.PYTHON_VER }}
           ARCH: ${{ matrix.ARCH }}
+          IMAGE_REPO: ${{ inputs.IMAGE_REPO }}
       - name: Build image
         uses: docker/build-push-action@v5
         with:
diff --git a/ci/compute-build-args.sh b/ci/compute-build-args.sh
index 3ee3313..7a1087d 100755
--- a/ci/compute-build-args.sh
+++ b/ci/compute-build-args.sh
@@ -1,15 +1,20 @@
 #!/bin/bash
 set -euo pipefail
 
-MANYLINUX_VER="manylinux_2_17"
+if [[
+  "${IMAGE_REPO}" == "ci-wheel" &&
+  "${LINUX_VER}" != "ubuntu20.04" &&
+  "${LINUX_VER}" != "rockylinux8"
+]]; then
+  echo "Unsupported LINUX_VER: ${LINUX_VER} for ci-wheel image"
+  exit 1
+fi
+
+MANYLINUX_VER="manylinux_2_28"
 if [[
   "${LINUX_VER}" == "ubuntu20.04"
 ]]; then
   MANYLINUX_VER="manylinux_2_31"
-elif [[
-  "${LINUX_VER}" == "rockylinux8"
-]]; then
-  MANYLINUX_VER="manylinux_2_28"
 fi
 
 ARGS="