Skip to content

Commit

Permalink
minimum torch 2.2.2
Browse files Browse the repository at this point in the history
  • Loading branch information
casper-hansen committed Nov 16, 2024
1 parent 887d08b commit 695382a
Showing 1 changed file with 34 additions and 32 deletions.
66 changes: 34 additions & 32 deletions .github/workflows/build.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -2,40 +2,40 @@ name: Build AutoAWQ Wheels with CUDA

on:
push:
tags:
- "v*"
branches:
- "downgrade_build_torch"

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_wheels:
name: Build AWQ
runs-on: ${{ matrix.os }}
needs: release
# needs: release

strategy:
matrix:
Expand All @@ -44,6 +44,8 @@ jobs:
defaults:
run:
shell: pwsh
env:
MIN_TORCH_VER: "2.2.0"

steps:
- name: Free Disk Space
Expand Down Expand Up @@ -78,7 +80,7 @@ jobs:
- name: Install Dependencies
run: |
# Install torch
python -m pip install --upgrade --no-cache-dir torch==2.5.1
python -m pip install --upgrade --no-cache-dir torch==$env:MIN_TORCH_VER
python -m pip install build setuptools wheel
# Print version information
Expand All @@ -89,8 +91,8 @@ jobs:
run: |
python setup.py sdist bdist_wheel
- name: Upload Assets
uses: shogo82148/actions-upload-release-asset@v1
with:
upload_url: ${{ needs.release.outputs.upload_url }}
asset_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

0 comments on commit 695382a

Please sign in to comment.