[build_tools] Add cmake preset for clang python wheel release (#559)

This PR adds cmake preset for clang python wheel release and updates
`build_wheels.sh` script to use this preset.

Author: shelkesagar29

mlir-tensorrt/CMakePresets.json

@@ -70,6 +70,23 @@
         "CMAKE_C_COMPILER": "gcc",
         "CMAKE_CXX_COMPILER": "g++"
       }
+    },
+    {
+      "name": "ninja-clang-wheel-release",
+      "displayName": "Ninja Release clang wheels",
+      "generator": "Ninja",
+      "binaryDir": "build",
+      "inherits": "base",
+      "cacheVariables": {
+        "CMAKE_C_COMPILER": "clang",
+        "CMAKE_CXX_COMPILER": "clang++",
+        "LLVM_ENABLE_ASSERTIONS": "OFF",
+        "CMAKE_BUILD_TYPE": "Release",
+        "CMAKE_PLATFORM_NO_VERSIONED_SONAME": "ON",
+        "LLVM_MINIMUM_PYTHON_VERSION": "$env{PY_VERSION}",
+        "MLIR_TRT_ENABLE_NCCL": "OFF",
+        "MLIR_TRT_DOWNLOAD_TENSORRT_VERSION": "$env{DOWNLOAD_TENSORRT_VERSION}"
+      }
     }
   ]
 }

mlir-tensorrt/build_tools/scripts/build_wheels.sh

@@ -1,25 +1,23 @@
 #!/usr/bin/env bash
 # Builds Python wheels for MLIR-TensorRT packages using the specified python version number.
-# Usage: build_wheels.sh 3.X
+# Usage:
+# build_wheels.sh [python version] [TensorRT version]
+# e.g. build_wheels.sh 3.X 10.X
 set -e
-version=${1:-3.10}
-PATH=$PATH:/pyenv/bin
+py_version=${PY_VERSION:-${1:-3.10}}
 
+PATH=$PATH:/pyenv/bin
 mkdir -p .private.wheels || true
+pyenv local ${py_version}
+python${py_version} -m pip install -r python/requirements-dev.txt
 
-pyenv local ${version}
-
-python${version} -m pip install -r python/requirements-dev.txt
+# If `DOWNLOAD_TENSORRT_VERSION` is not set, use the second argument as the version number.
+if [ -z "${DOWNLOAD_TENSORRT_VERSION}" ]; then
+    DOWNLOAD_TENSORRT_VERSION=$2
+    export DOWNLOAD_TENSORRT_VERSION
+fi
 
 rm -rf build || true
-cmake -B ./build -S . -G Ninja \
-    -DCMAKE_BUILD_TYPE=Release \
-    -DMLIR_TRT_PACKAGE_CACHE_DIR=${PWD}/.cache.cpm \
-    -DCMAKE_PLATFORM_NO_VERSIONED_SONAME=ON -DLLVM_MINIMUM_PYTHON_VERSION=${version} \
-    -DMLIR_TRT_ENABLE_ASSERTIONS=OFF \
-    -DMLIR_TRT_ENABLE_NCCL=OFF \
-    -DMLIR_TRT_DOWNLOAD_TENSORRT_VERSION=10.2
-
+cmake --preset ninja-clang-wheel-release
 ninja -C build mlir-tensorrt-all-wheels
-
 rsync -za build/wheels/ .private.wheels/

mlir-tensorrt/python/requirements-dev.txt

@@ -2,7 +2,6 @@
 # The GPU version installs `nvidia-` pip packages which confuse JAX.
 -r requirements.txt
 --extra-index-url https://download.pytorch.org/whl/cpu
-numpy<2.0.0
 click
 matplotlib
 pandas