From 802d59d0cda7f8007d8c8a2ce3b40341d24ea99e Mon Sep 17 00:00:00 2001 From: Tri Dao Date: Thu, 5 Dec 2024 23:04:05 -0800 Subject: [PATCH] [CI] Set wheel name with cu12 instead of cu124 --- .github/workflows/publish.yaml | 2 +- causal_conv1d/__init__.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/publish.yaml b/.github/workflows/publish.yaml index 9c2a51b..cb7561d 100644 --- a/.github/workflows/publish.yaml +++ b/.github/workflows/publish.yaml @@ -150,7 +150,7 @@ jobs: export LD_LIBRARY_PATH=/usr/local/nvidia/lib64:/usr/local/cuda/lib64:$LD_LIBRARY_PATH # Limit MAX_JOBS otherwise the github runner goes OOM MAX_JOBS=2 CAUSAL_CONV1D_FORCE_BUILD="TRUE" CAUSAL_CONV1D_FORCE_CXX11_ABI=${{ matrix.cxx11_abi}} python setup.py bdist_wheel --dist-dir=dist - tmpname=cu${MATRIX_CUDA_VERSION}torch${MATRIX_TORCH_VERSION}cxx11abi${{ matrix.cxx11_abi }} + tmpname=cu${WHEEL_CUDA_VERSION}torch${MATRIX_TORCH_VERSION}cxx11abi${{ matrix.cxx11_abi }} wheel_name=$(ls dist/*whl | xargs -n 1 basename | sed "s/-/+$tmpname-/2") ls dist/*whl |xargs -I {} mv {} dist/${wheel_name} echo "wheel_name=${wheel_name}" >> $GITHUB_ENV diff --git a/causal_conv1d/__init__.py b/causal_conv1d/__init__.py index 7629e3a..2182dc3 100644 --- a/causal_conv1d/__init__.py +++ b/causal_conv1d/__init__.py @@ -1,3 +1,3 @@ -__version__ = "1.5.0.post4" +__version__ = "1.5.0.post5" from causal_conv1d.causal_conv1d_interface import causal_conv1d_fn, causal_conv1d_update