Attempt fix

Author: casper-hansen

.github/workflows/build.yaml

@@ -2,40 +2,42 @@ name: Build AutoAWQ Wheels with CUDA
 
 on:
   push:
-    tags:
-      - "v*"
+    branches:
+      - "fix_workflow"
+    # 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
+  # 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: 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)
+  #     - 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:
@@ -113,17 +115,29 @@ jobs:
           }
 
           python setup.py sdist bdist_wheel
-      
-      - name: Upload Assets
-        uses: shogo82148/actions-upload-release-asset@v1
+
+      - 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
+
+      - uses: actions/upload-artifact@v3
+        if: runner.os == 'Windows'
+        with:
+          name: 'windows-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:
@@ -225,8 +239,20 @@ jobs:
 
           ROCM_VERSION=${{ matrix.rocm }} python setup.py sdist bdist_wheel
 
-      - name: Upload Assets
-        uses: shogo82148/actions-upload-release-asset@v1
+      - uses: actions/upload-artifact@v3
+        if: runner.os == 'Linux'
+        with:
+          name: 'linux-cuda-wheels'
+          path: ./dist/*.whl
+
+      - 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 Assets
+      #   uses: shogo82148/actions-upload-release-asset@v1
+      #   with:
+      #     upload_url: ${{ needs.release.outputs.upload_url }}
+      #     asset_path: ./dist/*.whl

setup.py

@@ -2,7 +2,6 @@
 import torch
 import platform
 import requests
-import importlib_metadata
 from pathlib import Path
 from setuptools import setup, find_packages
 
@@ -94,9 +93,13 @@ def get_kernels_whl_url(
 ]
 
 try:
-    importlib_metadata.version("autoawq-kernels")
+    if ROCM_VERSION:
+        import exlv2_ext
+    else:
+        import awq_ext
+
     KERNELS_INSTALLED = True
-except importlib_metadata.PackageNotFoundError:
+except ImportError:
     KERNELS_INSTALLED = False
 
 # kernels can be downloaded from pypi for cuda+121 only