test

Author: lanluo-nvidia

.github/workflows/mlir-tensorrt-build-test.yml

@@ -4,7 +4,7 @@ on:
   workflow_dispatch:
   pull_request:
   push:
-    branches: 
+    branches:
       - main
       - nightly
       - release/*
@@ -18,19 +18,6 @@ jobs:
     steps:
       - name: checkout TensorRT-Incubator
         uses: actions/checkout@v4
-      - name: python format and clang check
-        run: |
-          set -eou pipefail
-          set -x
-          python3 -m pip install black
-          python3 -m black --check --extend-exclude='.*\.pyi' mlir-tensorrt/compiler/
-          python3 -m black --check --extend-exclude='.*\.pyi' mlir-tensorrt/integrations/python/
-          git clang-format HEAD~1 --diff
-          if [ $? -ne 0 ]; then
-            echo "Python format and clang check failed"
-            exit 1
-          fi
-          echo "Python format and clang check passed"
       - name: Generate matrix
         id: generate
         run: |
@@ -46,12 +33,32 @@ jobs:
       matrix:
         ${{ fromJson(needs.generate-matrix.outputs.matrix) }}
     runs-on: [linux-amd64-gpu-h100-latest-1]
-    container: 
+    container:
       image: ${{ matrix.docker_image }}
       options: '--gpus all'
     steps:
+      - name: clean workspace
+        run: |
+          set -eou pipefail
+          set -x
+          echo "cleaning up ${GITHUB_WORKSPACE}"
+          rm -rf ${GITHUB_WORKSPACE}
+          mkdir -p ${GITHUB_WORKSPACE}
       - name: checkout TensorRT-Incubator
         uses: actions/checkout@v4
+      - name: python format and clang check
+        run: |
+          set -eou pipefail
+          set -x
+          python3 -m pip install black
+          python3 -m black --check --extend-exclude='.*\.pyi' mlir-tensorrt/compiler/
+          python3 -m black --check --extend-exclude='.*\.pyi' mlir-tensorrt/integrations/python/
+          git clang-format HEAD~1 --diff
+          if [ $? -ne 0 ]; then
+            echo "Python format and clang check failed"
+            exit 1
+          fi
+          echo "Python format and clang check passed"
       - name: create cache folder
         run: |
           set -eou pipefail
@@ -108,4 +115,7 @@ jobs:
           cd mlir-tensorrt
           python3 -m pip install .private.wheels/mlir_tensorrt-*.whl
           # TODO: add smoke test for the wheel
-          
+
+concurrency:
+  group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.ref_name }}-tensorrt-${{ inputs.repository }}-${{ github.event_name == 'workflow_dispatch' }}-${{ inputs.job-name }}
+  cancel-in-progress: true

mlir-tensorrt/.pre-commit-config.yaml

@@ -0,0 +1,70 @@
+exclude: ^.github/actions/assigner/dist
+repos:
+    - repo: https://github.com/pre-commit/pre-commit-hooks
+      rev: v5.0.0
+      hooks:
+          - id: check-yaml
+          - id: trailing-whitespace
+            exclude: ^docs
+          - id: check-added-large-files
+            args:
+                - --maxkb=1000
+          - id: check-vcs-permalinks
+          - id: check-merge-conflict
+          - id: mixed-line-ending
+            args:
+                - --fix=lf
+            exclude: ^docs
+    - repo: https://github.com/pre-commit/mirrors-clang-format
+      rev: v14.0.6
+      hooks:
+          - id: clang-format
+            types_or: [c++, c, cuda]
+    - repo: https://github.com/keith/pre-commit-buildifier
+      rev: 8.0.3
+      hooks:
+          - id: buildifier
+            args:
+                - --warnings=all
+          - id: buildifier-lint
+    - repo: https://github.com/abravalheri/validate-pyproject
+      rev: v0.24.1
+      hooks:
+          - id: validate-pyproject
+    - repo: https://github.com/pycqa/isort
+      rev: 6.0.0
+      hooks:
+          - id: isort
+            name: isort (python)
+    - repo: https://github.com/pre-commit/mirrors-mypy
+      rev: "v1.15.0"
+      hooks:
+          - id: mypy
+            files: "^(integrations/python|tensorrt/python)/.*\\.py$"
+    - repo: https://github.com/astral-sh/ruff-pre-commit
+      # Ruff version.
+      rev: v0.11.7
+      hooks:
+          - id: ruff
+    - repo: https://github.com/psf/black
+      rev: 25.1.0
+      hooks:
+          - id: black
+    - repo: https://github.com/crate-ci/typos
+      rev: typos-dict-v0.12.4
+      hooks:
+          - id: typos
+    - repo: https://github.com/astral-sh/uv-pre-commit
+      # uv version.
+      rev: 0.7.1
+      hooks:
+          # Update the uv lockfile
+          - id: uv-lock
+    - repo: local
+      hooks:
+          - id: dont-commit-upstream
+            name: NVIDIA-INTERNAL check
+            entry: "!NVIDIA-INTERNAL"
+            exclude: "mlir-tensorrt/.pre-commit-config.yaml"
+            language: pygrep
+            types: [text]