update

Author: MayDomine

.github/workflows/build_whl.yml

@@ -44,14 +44,36 @@ jobs:
         version=${{ matrix.python-version }}
         docker run -e BUILD_DOCKER_ENV=1 -e CUDACXX=/usr/local/cuda-11.3/bin/nvcc -e PATH="/opt/rh/devtoolset-9/root/usr/bin:$PATH" -e LD_LIBRARY_PATH="/opt/rh/devtoolset-9/root/usr/lib64:/opt/rh/devtoolset-9/root/usr/lib:/usr/local/nvidia/lib:/usr/local/nvidia/lib64:$LD_LIBRARY_PATH" -v ${{ github.workspace }}:/workspace/BMTrain -i pytorch/manylinux-cuda113:latest /bin/bash -c "cd /workspace/BMTrain;/opt/python/cp${version}*/bin/pip install build; /opt/python/cp${version}*/bin/python -m build .;for file in dist/*-linux_x86_64.whl; do mv \"\$file\" \"\${file//-linux_x86_64/-manylinux2014_x86_64}\"; done"
 
+    - name: Upload wheels as artifacts
+      uses: actions/upload-artifact@v4
+      with:
+        name: wheels_py${{ matrix.python-version }}
+        path: |
+          dist/*.whl
+          dist/*.tar.gz
+    
+  archive:
+    runs-on: ubuntu-latest
+    needs: build-wheels
+    steps:
+    - name: Download all wheels
+      uses: actions/download-artifact@v4
+      with:
+        path: dist
+        pattern: |
+          wheels_py*/*.whl
+          wheels_py37/*.tar.gz
+
+    - name: Combine all wheels into a single directory
+      run: |
+        mkdir -p dist
+        find . -path './dist' -prune -o -name '*.whl' -exec mv {} dist/ \;
+
     - name: Archive distribution files
       uses: actions/upload-artifact@v4
       with:
         name: dist
         path: |
           dist/*.tar.gz
           dist/*.whl
-        overwrite: true
-
-
-    
+        overwrite: true