test

Author: lanluo-nvidia

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

@@ -42,8 +42,6 @@ jobs:
       packages: write
       contents: read
     strategy:
-      fail-fast: true
-      max-parallel: 1
       matrix: ${{ fromJSON(needs.generate-matrix.outputs.matrix) }}
     runs-on: ${{ matrix.github_runner }}
     env:
@@ -141,15 +139,30 @@ jobs:
           path: mlir-tensorrt/dist
           if-no-files-found: error
 
-      - name: Upload wheels to test PyPI
+  pypi-release-wheels-publish:
+    name: Publish to TestPyPI
+    needs: [pypi-release-wheels-build]
+    runs-on: ubuntu-latest
+    permissions:
+      id-token: write
+      packages: write
+      contents: read
+    steps:
+      - name: Download built wheels
+        uses: actions/download-artifact@v4
+        with:
+          pattern: release-wheels-*
+          merge-multiple: true
+          path: dist-collect
+
+      - name: Publish to TestPyPI
         uses: pypa/gh-action-pypi-publish@release/v1
         with:
-          packages-dir: mlir-tensorrt/dist
+          packages-dir: dist-collect
           repository-url: https://test.pypi.org/legacy/
           user: __token__
           password: ${{ secrets.TEST_PYPI_API_TOKEN }}
 
-
 concurrency:
   group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.ref_name }}-mlir-tensorrt-pypi
   cancel-in-progress: true