test

Author: lanluo-nvidia

.github/workflows/mlir-tensorrt-release.yml

@@ -8,7 +8,7 @@ on:
         default: ""
         type: string
       pypi-release:
-        description: 'Whether to build pypi wheels'
+        description: 'Flag to build and publish pypi wheels'
         default: false
         type: boolean
 
@@ -209,7 +209,7 @@ jobs:
           generate_release_notes: false
           draft: true
           prerelease: true
-      
+
       - name: Upload wheels to test PyPI
         if: ${{ inputs.pypi-release == true }}
         uses: pypa/gh-action-pypi-publish@release/v1
@@ -218,7 +218,7 @@ jobs:
           repository-url: https://test.pypi.org/legacy/
           user: __token__
           password: ${{ secrets.TEST_PYPI_API_TOKEN }}
-  
+
   # pypi-publish:
   #   name: Publish to PyPI (requires approval)
   #   needs: mlir-tensorrt-wheel-build

mlir-tensorrt/integrations/python/setup_utils.py

@@ -2,16 +2,17 @@
 Utility functions used in the setup.py files for the Python packages.
 """
 
+import atexit
 import datetime
 import os
 import re
 import shutil
+import subprocess
 import sys
-from pathlib import Path
 import tempfile
+from pathlib import Path
+
 import setuptools
-import subprocess
-import atexit
 
 TENSORRT_VERSION = os.getenv("MLIR_TRT_DOWNLOAD_TENSORRT_VERSION", "10.12")
 
@@ -63,10 +64,12 @@ def get_nightly_version() -> str:
     datestring = datetime.date.today().strftime("%Y%m%d")
     return append_version_feature_flags(f"{get_base_version()}.dev{datestring}")
 
+
 def get_pypi_version() -> str:
     # wheel upload to pypi need to follow the version specification in https://packaging.python.org/en/latest/specifications/core-metadata/
     return f"{get_base_version()}"
 
+
 def cleanup_dir(dir: Path, should_cleanup: bool, comment: str = ""):
     prefix = "Cleaning up" if should_cleanup else "Not cleaning up"
     log(f"{prefix} {comment} at {dir}")
@@ -157,7 +160,7 @@ def run_cmake_build(python_package_name: str, python_wheel_staging_dir: Path):
     log(f"Building MLIR-TensorRT in {build_dir}")
     log(f"Installing to {install_dir}")
 
-    # Retrieve the path to the isolated Python site where the wheel build is occuring.
+    # Retrieve the path to the isolated Python site where the wheel build is occurring.
     # That is where the Python build dependencies (the packages listed in
     # 'build-requires' in the pyproject.toml) are available. The actual CMake build
     # may invoke the Python interpreter (in order to e.g. generate custom MLIR files
@@ -210,7 +213,7 @@ def run_cmake_build(python_package_name: str, python_wheel_staging_dir: Path):
     subprocess.check_call(build_cmd, cwd=project_root)
 
     # Install
-    install_cmd = ["cmake", "--install", str(build_dir), "--strip"]
+    install_cmd = ["cmake", "--install", str(build_dir)]
 
     log(f"Install command: {' '.join(install_cmd)}")
     subprocess.check_call(install_cmd, cwd=project_root)