Bump release, test build

casper-hansen · Jan 26, 2024 · 41e3e11 · 41e3e11
1 parent 3aed4bf
commit 41e3e11
Show file tree

Hide file tree

Showing 2 changed files with 68 additions and 39 deletions.
diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml
@@ -2,40 +2,42 @@ name: Build AutoAWQ Wheels with CUDA
 
 on:
   push:
-    tags:
-      - "v*"
+    branches:
+      - "new_release"
+    # tags:
+    #   - "v*"
 
 jobs:
-  release:
-    # Retrieve tag and create release
-    name: Create Release
-    runs-on: ubuntu-latest
-    outputs:
-      upload_url: ${{ steps.create_release.outputs.upload_url }}
-    steps:
-      - name: Checkout
-        uses: actions/checkout@v3
-
-      - name: Extract branch info
-        shell: bash
-        run: |
-          echo "release_tag=${GITHUB_REF#refs/*/}" >> $GITHUB_ENV
-
-      - name: Create Release
-        id: create_release
-        uses: "actions/github-script@v6"
-        env:
-          RELEASE_TAG: ${{ env.release_tag }}
-        with:
-          github-token: "${{ secrets.GITHUB_TOKEN }}"
-          script: |
-            const script = require('.github/workflows/scripts/github_create_release.js')
-            await script(github, context, core)
+  # release:
+  #   # Retrieve tag and create release
+  #   name: Create Release
+  #   runs-on: ubuntu-latest
+  #   outputs:
+  #     upload_url: ${{ steps.create_release.outputs.upload_url }}
+  #   steps:
+  #     - name: Checkout
+  #       uses: actions/checkout@v3
+
+  #     - name: Extract branch info
+  #       shell: bash
+  #       run: |
+  #         echo "release_tag=${GITHUB_REF#refs/*/}" >> $GITHUB_ENV
+
+  #     - name: Create Release
+  #       id: create_release
+  #       uses: "actions/github-script@v6"
+  #       env:
+  #         RELEASE_TAG: ${{ env.release_tag }}
+  #       with:
+  #         github-token: "${{ secrets.GITHUB_TOKEN }}"
+  #         script: |
+  #           const script = require('.github/workflows/scripts/github_create_release.js')
+  #           await script(github, context, core)
 
   build_cuda_wheels:
     name: Build AWQ with CUDA
     runs-on: ${{ matrix.os }}
-    needs: release
+    # needs: release
 
     strategy:
       matrix:
@@ -114,16 +116,29 @@ jobs:
 
           python setup.py sdist bdist_wheel
 
-      - name: Upload Assets
-        uses: shogo82148/actions-upload-release-asset@v1
+      - name: Upload
+        uses: actions/upload-artifact@v3
+        if: runner.os == 'Windows'
         with:
-          upload_url: ${{ needs.release.outputs.upload_url }}
-          asset_path: ./dist/*.whl
+          name: 'windows-cuda-wheels'
+          path: ./dist/*.whl
+
+      - name: Upload
+        uses: actions/upload-artifact@v3
+        if: runner.os == 'Linux'
+        with:
+          name: 'linux-cuda-wheels'
+          path: ./dist/*.whl
+      # - name: Upload Assets
+      #   uses: shogo82148/actions-upload-release-asset@v1
+      #   with:
+      #     upload_url: ${{ needs.release.outputs.upload_url }}
+      #     asset_path: ./dist/*.whl
 
   build_rocm_wheels:
     name: Build AWQ with ROCm
     runs-on: ${{ matrix.os }}
-    needs: release
+    # needs: release
 
     strategy:
       matrix:
@@ -226,9 +241,23 @@ jobs:
           which python
 
           ROCM_VERSION=${{ matrix.rocm }} python setup.py sdist bdist_wheel
-
-      - name: Upload Assets
-        uses: shogo82148/actions-upload-release-asset@v1
+      
+      - name: Upload
+        uses: actions/upload-artifact@v3
+        if: runner.os == 'Windows'
+        with:
+          name: 'windows-cuda-wheels'
+          path: ./dist/*.whl
+
+      - name: Upload
+        uses: actions/upload-artifact@v3
+        if: runner.os == 'Linux'
         with:
-          upload_url: ${{ needs.release.outputs.upload_url }}
-          asset_path: ./dist/*.whl
+          name: 'linux-cuda-wheels'
+          path: ./dist/*.whl
+
+      # - name: Upload Assets
+      #   uses: shogo82148/actions-upload-release-asset@v1
+      #   with:
+      #     upload_url: ${{ needs.release.outputs.upload_url }}
+      #     asset_path: ./dist/*.whl
diff --git a/setup.py b/setup.py
@@ -7,7 +7,7 @@
 
 os.environ["CC"] = "g++"
 os.environ["CXX"] = "g++"
-AUTOAWQ_KERNELS_VERSION = "0.0.2"
+AUTOAWQ_KERNELS_VERSION = "0.0.3"
 PYPI_BUILD = os.getenv("PYPI_BUILD", "0") == "1"
 CUDA_VERSION = os.getenv("CUDA_VERSION", None) or torch.version.cuda
 ROCM_VERSION = os.environ.get("ROCM_VERSION", None) or torch.version.hip