[CI] Install libcublas and libcusparse

Dao-AILab · Dec 6, 2024 · 5d39d51 · 5d39d51
1 parent 7f9c423
commit 5d39d51
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/.github/workflows/publish.yaml b/.github/workflows/publish.yaml
@@ -109,6 +109,7 @@ jobs:
           # We need the cuda libraries (e.g. cuSparse, cuSolver) for compiling PyTorch extensions,
           # not just nvcc
           sub-packages: '["nvcc"]'
+          non-cuda-sub-packages: '["libcublas", "libcusparse"]'
 
       - name: Install PyTorch ${{ matrix.torch-version }}+cu${{ matrix.cuda-version }}
         run: |

diff --git a/causal_conv1d/__init__.py b/causal_conv1d/__init__.py
@@ -1,3 +1,3 @@
-__version__ = "1.5.0.post2"
+__version__ = "1.5.0.post3"
 
 from causal_conv1d.causal_conv1d_interface import causal_conv1d_fn, causal_conv1d_update