diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml index 8059cf1..0313fb9 100644 --- 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 index 89c9869..461878f 100644 --- 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