Skip to content

Commit 009bec5

Browse files
committed
[CI] Fix CUDA 12.2 compilation
1 parent 35689e1 commit 009bec5

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

.github/workflows/publish.yaml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -64,11 +64,11 @@ jobs:
6464
python-version: '3.7'
6565
# Pytorch <= 2.0 only supports CUDA <= 11.8
6666
- torch-version: '1.12.1'
67-
cuda-version: '12.2.0'
67+
cuda-version: '12.2.2'
6868
- torch-version: '1.13.1'
69-
cuda-version: '12.2.0'
69+
cuda-version: '12.2.2'
7070
- torch-version: '2.0.1'
71-
cuda-version: '12.2.0'
71+
cuda-version: '12.2.2'
7272

7373
steps:
7474
- name: Checkout

mamba_ssm/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
__version__ = "1.1.3"
1+
__version__ = "1.1.3.post1"
22

33
from mamba_ssm.ops.selective_scan_interface import selective_scan_fn, mamba_inner_fn
44
from mamba_ssm.modules.mamba_simple import Mamba

0 commit comments

Comments
 (0)