Skip to content

Commit

Permalink
[CI] Fix CUDA 12.2 compilation
Browse files Browse the repository at this point in the history
  • Loading branch information
tridao committed Feb 1, 2024
1 parent a30eae4 commit d1ab53d
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 4 deletions.
6 changes: 3 additions & 3 deletions .github/workflows/publish.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
2 changes: 1 addition & 1 deletion causal_conv1d/__init__.py
Original file line number Diff line number Diff line change
@@ -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

0 comments on commit d1ab53d

Please sign in to comment.