From a412bdf90a6c8d465304fa80890e28b3f7a34853 Mon Sep 17 00:00:00 2001
From: Vadim Gimpelson <156319763+vadiklyutiy@users.noreply.github.com>
Date: Tue, 30 Jul 2024 06:39:26 +0400
Subject: [PATCH] [BUILD] Build wheel with `manylinux_2_28_x86_64` (#471)

- build wheel with `manylinux_2_28_x86_64`
- v0.4.0 -> v0.4.1
---
 .github/actions/action.yaml                                | 2 +-
 python/hidet/version.py                                    | 2 +-
 scripts/wheel/build_wheel.sh                               | 2 +-
 scripts/wheel/dockerfiles/manylinux1/Dockerfile            | 2 +-
 scripts/wheel/dockerfiles/manylinux_2_28_x86_64/Dockerfile | 6 ++++++
 setup.py                                                   | 2 +-
 6 files changed, 11 insertions(+), 5 deletions(-)
 create mode 100644 scripts/wheel/dockerfiles/manylinux_2_28_x86_64/Dockerfile

diff --git a/.github/actions/action.yaml b/.github/actions/action.yaml
index 889f9b281..c0647b25d 100644
--- a/.github/actions/action.yaml
+++ b/.github/actions/action.yaml
@@ -9,7 +9,7 @@ outputs:
     description: 'the path to the generated wheel'
 runs:
   using: 'docker'
-  image: '../../scripts/wheel/dockerfiles/manylinux1/Dockerfile'
+  image: '../../scripts/wheel/dockerfiles/manylinux_2_28_x86_64/Dockerfile'
   args:
     - "bash"
     - "./scripts/wheel/build_wheel.sh"
diff --git a/python/hidet/version.py b/python/hidet/version.py
index 37227dafa..8d438b1ee 100644
--- a/python/hidet/version.py
+++ b/python/hidet/version.py
@@ -9,4 +9,4 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-__version__ = "0.4.0"
+__version__ = "0.4.1"
diff --git a/scripts/wheel/build_wheel.sh b/scripts/wheel/build_wheel.sh
index 0f60cb160..5ff47c775 100644
--- a/scripts/wheel/build_wheel.sh
+++ b/scripts/wheel/build_wheel.sh
@@ -40,7 +40,7 @@ fi
 
 # build wheel
 mkdir -p built_wheel;
-cd built_wheel; pip wheel --no-deps ..; cd ..
+cd built_wheel; pip3 wheel --no-deps ..; cd ..
 
 # remove all intermediate directories
 rm -rf ./python
diff --git a/scripts/wheel/dockerfiles/manylinux1/Dockerfile b/scripts/wheel/dockerfiles/manylinux1/Dockerfile
index febde290f..5c57fb774 100644
--- a/scripts/wheel/dockerfiles/manylinux1/Dockerfile
+++ b/scripts/wheel/dockerfiles/manylinux1/Dockerfile
@@ -1,4 +1,4 @@
-FROM quay.io/pypa/manylinux_2_28_x86_64 as base
+FROM quay.io/pypa/manylinux1_x86_64 as base
 
 RUN yum install -y vim htop
 
diff --git a/scripts/wheel/dockerfiles/manylinux_2_28_x86_64/Dockerfile b/scripts/wheel/dockerfiles/manylinux_2_28_x86_64/Dockerfile
new file mode 100644
index 000000000..970e6024b
--- /dev/null
+++ b/scripts/wheel/dockerfiles/manylinux_2_28_x86_64/Dockerfile
@@ -0,0 +1,6 @@
+FROM quay.io/pypa/manylinux_2_28_x86_64 as base
+
+# Install pip
+RUN python3 -m ensurepip --upgrade
+
+RUN pip3 install wheel
\ No newline at end of file
diff --git a/setup.py b/setup.py
index c1ec392cd..875853fe3 100644
--- a/setup.py
+++ b/setup.py
@@ -14,7 +14,7 @@
 
 setup(
     name="hidet",
-    version="0.4.0",
+    version="0.4.1",
     description="Hidet: a compilation-based DNN inference framework.",
     long_description=open("README.md").read(),
     long_description_content_type="text/markdown",