[ci] Fixing releasing (#8651)

Author: feisuzhu

.github/workflows/release.yml

@@ -97,18 +97,24 @@ jobs:
           echo CI_DOCKER_RUN_EXTRA_ARGS="-v $(pwd):/home/dev/taichi" >> $GITHUB_ENV
 
       - name: Build
+        id: build
         run: |
           . .github/workflows/scripts/common-utils.sh
 
           ci-docker-run-gpu --name taichi-build \
             registry.botmaster.tgr/taichi-build-cuda:${{ env.CI_IMAGE_VERSION }} \
             /home/dev/taichi/.github/workflows/scripts/build.py $NIGHTLY
 
+          # Get wheel filename
+          WHEEL_NAME=$(ls dist/*.whl | head -n 1 | xargs basename)
+          echo "wheel_name=$WHEEL_NAME" >> $GITHUB_OUTPUT
+          echo "Found wheel: $WHEEL_NAME"
+
       - name: Archive Wheel Artifacts
         uses: actions/upload-artifact@v4
         with:
-          name: dist
-          path: dist/*.whl
+          name: ${{ steps.build.outputs.wheel_name }}
+          path: dist/${{ steps.build.outputs.wheel_name }}
           retention-days: 20
 
       - name: Test
@@ -157,8 +163,8 @@ jobs:
   #     - name: Archive Wheel Artifacts
   #       uses: actions/upload-artifact@v4
   #       with:
-  #         name: dist
-  #         path: dist/*.whl
+  #         name: ${{ steps.build.outputs.wheel_name }}
+  #         path: dist/${{ steps.build.outputs.wheel_name }}
   #         retention-days: 20
 
   #     - name: Test
@@ -199,15 +205,21 @@ jobs:
           prepare-build-cache
 
       - name: Build
+        id: build
         run: |
           brew install molten-vk
           .github/workflows/scripts/build.py $NIGHTLY
 
+          # Get wheel filename
+          WHEEL_NAME=$(ls dist/*.whl | head -n 1 | xargs basename)
+          echo "wheel_name=$WHEEL_NAME" >> $GITHUB_OUTPUT
+          echo "Found wheel: $WHEEL_NAME"
+
       - name: Archive Wheel Artifacts
         uses: actions/upload-artifact@v4
         with:
-          name: dist
-          path: dist/*.whl
+          name: ${{ steps.build.outputs.wheel_name }}
+          path: dist/${{ steps.build.outputs.wheel_name }}
           retention-days: 20
 
       - name: Test
@@ -253,6 +265,7 @@ jobs:
           python-version: ${{ matrix.python }}
 
       - name: Build
+        id: build
         shell: pwsh
         run: |
           $nightlyFlag = $null
@@ -261,11 +274,16 @@ jobs:
           SetGitCachingProxy
           python .\.github\workflows\scripts\build.py $nightlyFlag
 
+          # Get wheel filename
+          $WheelName = (Get-ChildItem -Path dist\*.whl | Select-Object -First 1).Name
+          echo "wheel_name=$WheelName" >> $env:GITHUB_OUTPUT
+          echo "Found wheel: $WheelName"
+
       - name: Archive Wheel Artifacts
         uses: actions/upload-artifact@v4
         with:
-          name: dist
-          path: dist/*.whl
+          name: ${{ steps.build.outputs.wheel_name }}
+          path: dist/${{ steps.build.outputs.wheel_name }}
           retention-days: 20
 
       - name: Test
@@ -297,15 +315,16 @@ jobs:
         with:
           python-version: 3.9
 
-      - name: Get dist files
+      - name: Download all artifacts
         uses: actions/download-artifact@v4
         with:
-          name: dist
-          path: dist
+          path: zips
 
       - name: Upload to PyPI
         run: |
           set -ex
+          mkdir -p dist
+          mv zips/*/*.whl dist/
           ls -l dist/
           if [ -z "$RELEASE_VERSION" ]; then
             export PROJECT_NAME="taichi-nightly"