debug wheels dir issue

Author: lanluo-nvidia

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

@@ -194,16 +194,16 @@ jobs:
       arch: x86_64
       github_runner: linux-amd64-gpu-h100-latest-1
 
-  mlir-tensorrt-wheel-tarball-release-aarch64:
-    name: Release on aarch64
-    needs:
-      - format-check
-    if: ${{ needs.format-check.outputs.channel == 'release' }}
-    uses: ./.github/workflows/mlir-tensorrt-release.yml
-    with:
-      build-matrix: ${{ needs.format-check.outputs.matrix }}
-      arch: aarch64
-      github_runner: linux-arm64-gpu-l4-latest-1
+  # mlir-tensorrt-wheel-tarball-release-aarch64:
+  #   name: Release on aarch64
+  #   needs:
+  #     - format-check
+  #   if: ${{ needs.format-check.outputs.channel == 'release' }}
+  #   uses: ./.github/workflows/mlir-tensorrt-release.yml
+  #   with:
+  #     build-matrix: ${{ needs.format-check.outputs.matrix }}
+  #     arch: aarch64
+  #     github_runner: linux-arm64-gpu-l4-latest-1
 
 concurrency:
   group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.ref_name }}-mlir-tensorrt

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

@@ -101,14 +101,16 @@ jobs:
           export MTRT_TENSORRT_VERSION=${{ matrix.trt }}
           # TODO: remove this, once patch is in our rockylinux prebuilt image
           dnf install -yq patch
-          # mlir-tensorrt-tools is not cuda, trt dependent, so we only build it once
-          # mlir-tensorrt-compiler, mlir-tensorrt-runtime is built per cuda, trt version
-          if [[ "${{ matrix.cuda }}" == "${{ env.LATEST_CUDA_VERSION }}" && "${{ matrix.trt }}" == "${{env.LATEST_TENSORRT_VERSION }}" ]]; then
-            ./build_tools/scripts/cicd-build-wheels.sh
-          else
-            PACKAGES="mlir_tensorrt_compiler mlir_tensorrt_runtime" \
-            ./build_tools/scripts/cicd-build-wheels.sh
-          fi
+          PACKAGES="mlir_tensorrt_tools" PYTHON_VERSIONS="3.12" ./build_tools/scripts/cicd-build-wheels.sh
+
+          # # mlir-tensorrt-tools is not cuda, trt dependent, so we only build it once
+          # # mlir-tensorrt-compiler, mlir-tensorrt-runtime is built per cuda, trt version
+          # if [[ "${{ matrix.cuda }}" == "${{ env.LATEST_CUDA_VERSION }}" && "${{ matrix.trt }}" == "${{env.LATEST_TENSORRT_VERSION }}" ]]; then
+          #   ./build_tools/scripts/cicd-build-wheels.sh
+          # else
+          #   PACKAGES="mlir_tensorrt_compiler mlir_tensorrt_runtime" \
+          #   ./build_tools/scripts/cicd-build-wheels.sh
+          # fi
 
       # Save ccache when cache is not hit or cache was a fallback(cache-matched is not the same as the cache key)
       - name: Save CCache
@@ -144,8 +146,8 @@ jobs:
         uses: actions/upload-artifact@v4
         with:
           name: release-wheels-${{ inputs.arch }}-cu${{ matrix.cuda }}-trt${{ matrix.trt }}
-          path: ${{ env.WHEELS_DIR }}
-          if-no-files-found: error
+          path: mlir-tensorrt/.wheels
+          if-no-files-found: warn
 
 concurrency:
   group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.ref_name }}-mlir-tensorrt-release-${{ inputs.arch }}

mlir-tensorrt/build_tools/scripts/cicd-build-wheels.sh

@@ -54,7 +54,6 @@ mkdir -p "${MLIR_TRT_BUILD_DIR}"
 export MLIR_TRT_INSTALL_DIR="${MLIR_TRT_INSTALL_DIR:-${REPO_ROOT}/install}"
 mkdir -p "${MLIR_TRT_INSTALL_DIR}"
 export WHEELS_DIR="${WHEELS_DIR:-${REPO_ROOT}/.wheels}"
-mkdir -p "${WHEELS_DIR}"
 
 export MTRT_TENSORRT_VERSION="${MTRT_TENSORRT_VERSION:-10.13}"
 
@@ -86,4 +85,5 @@ du -h -x -d 1 ${CPM_SOURCE_CACHE}
 echo "🔨🧪 Printing the size of the wheels directory"
 du -h -x -d 1 "${WHEELS_DIR}"
 
-ls -lart "${WHEELS_DIR}"
+ls -lart "${WHEELS_DIR}"
+ls -lart mlir-tensorrt/.wheels