update workflow

Author: MayDomine

.github/workflows/build.yml

@@ -13,10 +13,10 @@ on:
 jobs:
   build-archive-wheel:
 
-    uses: OpenBMB/BMTrain/.github/workflows/build_whl.yml@main
+    uses: OpenBMB/BMTrain/.github/workflows/build_whl.yml@dev
     secrets: inherit
 
-  publish:
+  fake-publish:
     needs: build-archive-wheel
     runs-on: ubuntu-latest
     steps:
@@ -29,7 +29,7 @@ jobs:
         python-version: 3.9
 
     - name: Download distribution files
-      uses: actions/download-artifact@v2
+      uses: actions/download-artifact@v4
       with:
         name: dist
         path: dist

.github/workflows/build_whl.yml

@@ -44,12 +44,46 @@ 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
+
+    - name: Upload source distribution (only once)
+      if: matrix.python-version == '37'  # Only upload source distribution once
+      uses: actions/upload-artifact@v4
+      with:
+        name: source_dist
+        path: dist/*.tar.gz
+    
+  archive:
+    runs-on: ubuntu-latest
+    needs: build
+    steps:
+    - name: Download all wheels
+      uses: actions/download-artifact@v4
+      with:
+        path: wheels
+        pattern: wheels_py*
+
+    - name: Download source distribution
+      uses: actions/download-artifact@v4
+      with:
+        path: dist
+        pattern: source_dist/*.tar.gz
+
+    - name: Combine all wheels into a single directory
+      run: |
+        mkdir -p dist
+        find wheels -name '*.whl' -exec mv {} dist/ \;
+        find dist -name '*.tar.gz' -exec mv {} dist/ \;
+
     - name: Archive distribution files
-      uses: actions/upload-artifact@v2
+      uses: actions/upload-artifact@v4
       with:
         name: dist
         path: |
           dist/*.tar.gz
           dist/*.whl
-
-    
+        overwrite: true

.github/workflows/release.yml

@@ -40,7 +40,7 @@ jobs:
           dist/*.tar.gz
           dist/*.whl
         prerelease: false
-        name: "BMTrain v0.2.3"
+        name: "BMTrain"
         token: ${{ secrets.RELEASE_TOKEN }}
         tag_name: ${{ steps.create_release.outputs.tag }}
       env: