address review comments

Signed-off-by: Anatoly Myachev <[email protected]>

Author: anmyachev

.github/workflows/integration-tests.yml

@@ -236,7 +236,7 @@ jobs:
       - name: Install pip dependencies
         run: |
           python3 -m pip install --upgrade pip
-          python3 -m pip install cython wheel cmake==3.24 ninja pytest-forked pytest-xdist lit
+          python3 -m pip install cython setuptools>=65.6.1 wheel cmake==3.24 ninja pytest-forked pytest-xdist lit
       - name: Install Triton
         env:
           CUDA_HOME: "/usr/local/cuda"
@@ -569,7 +569,7 @@ jobs:
           python3 -m venv ~/.venv
           source ~/.venv/bin/activate
           python3 -m pip install --upgrade pip
-          python3 -m pip install cython wheel cmake==3.24 ninja pytest-xdist lit pybind11
+          python3 -m pip install cython setuptools>=65.6.1 wheel cmake==3.24 ninja pytest-xdist lit pybind11
       - name: Install Triton
         env:
           TRITON_BUILD_WITH_O1: "true"

.github/workflows/integration-tests.yml.in

@@ -268,7 +268,7 @@ jobs:
       - name: Install pip dependencies
         run: |
           python3 -m pip install --upgrade pip
-          python3 -m pip install cython wheel cmake==3.24 ninja pytest-forked pytest-xdist lit
+          python3 -m pip install cython setuptools>=65.6.1 wheel cmake==3.24 ninja pytest-forked pytest-xdist lit
 
       - name: Install Triton
         env:
@@ -481,7 +481,7 @@ jobs:
           python3 -m venv ~/.venv
           source ~/.venv/bin/activate
           python3 -m pip install --upgrade pip
-          python3 -m pip install cython wheel cmake==3.24 ninja pytest-xdist lit pybind11
+          python3 -m pip install cython setuptools>=65.6.1 wheel cmake==3.24 ninja pytest-xdist lit pybind11
       - name: Install Triton
         env:
           TRITON_BUILD_WITH_O1: "true"

pyproject.toml

@@ -1,5 +1,5 @@
 [build-system]
-requires = ["setuptools>=60.8.0", "wheel", "cmake>=3.18", "ninja>=1.11.1"]
+requires = ["setuptools>=65.6.1", "wheel", "cmake>=3.18", "ninja>=1.11.1"]
 
 [tool.yapf]
 based_on_style = "pep8"

python/pyproject.toml

@@ -1,6 +1,6 @@
 
 [build-system]
-requires = ["setuptools>=60.8.0", "wheel", "cmake>=3.18", "ninja>=1.11.1", "pybind11>=2.13.1"]
+requires = ["setuptools>=65.6.1", "wheel", "cmake>=3.18", "ninja>=1.11.1", "pybind11>=2.13.1"]
 
 # We're incrementally switching from autopep8 to ruff.
 [tool.autopep8]

python/setup.py

@@ -765,7 +765,7 @@ def get_git_commit_hash(length=8):
     author_email="[email protected]",
     description="A language and compiler for custom Deep Learning operations",
     long_description="",
-    install_requires=["setuptools>=60.8.0"],
+    install_requires=["setuptools>=65.6.1"],
     packages=get_packages(),
     entry_points=get_entry_points(),
     package_data=package_data,