From d1ab53ddf93b868502da96ad82c5fc0b270911f5 Mon Sep 17 00:00:00 2001 From: Tri Dao Date: Wed, 31 Jan 2024 18:51:08 -0800 Subject: [PATCH] [CI] Fix CUDA 12.2 compilation --- .github/workflows/publish.yaml | 6 +++--- causal_conv1d/__init__.py | 2 +- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/.github/workflows/publish.yaml b/.github/workflows/publish.yaml index 4cd66b0..b3f8c41 100644 --- a/.github/workflows/publish.yaml +++ b/.github/workflows/publish.yaml @@ -64,11 +64,11 @@ jobs: python-version: '3.7' # Pytorch <= 2.0 only supports CUDA <= 11.8 - torch-version: '1.12.1' - cuda-version: '12.2.0' + cuda-version: '12.2.2' - torch-version: '1.13.1' - cuda-version: '12.2.0' + cuda-version: '12.2.2' - torch-version: '2.0.1' - cuda-version: '12.2.0' + cuda-version: '12.2.2' steps: - name: Checkout diff --git a/causal_conv1d/__init__.py b/causal_conv1d/__init__.py index 77d9895..a2af8e9 100644 --- a/causal_conv1d/__init__.py +++ b/causal_conv1d/__init__.py @@ -1,3 +1,3 @@ -__version__ = "1.1.2" +__version__ = "1.1.2.post1" from causal_conv1d.causal_conv1d_interface import causal_conv1d_fn, causal_conv1d_update