File tree Expand file tree Collapse file tree 2 files changed +4
-2
lines changed Expand file tree Collapse file tree 2 files changed +4
-2
lines changed Original file line number Diff line number Diff line change @@ -120,6 +120,7 @@ RUN <<EOT bash
120
120
apt-get update && \
121
121
apt-get install -y --no-install-recommends \
122
122
cuda-nvcc-${CUDA_MAJOR_VERSION}-${CUDA_MINOR_VERSION} \
123
+ libcufft-dev-${CUDA_MAJOR_VERSION}-${CUDA_MINOR_VERSION} \
123
124
libcurand-dev-${CUDA_MAJOR_VERSION}-${CUDA_MINOR_VERSION} \
124
125
libcublas-dev-${CUDA_MAJOR_VERSION}-${CUDA_MINOR_VERSION} \
125
126
libcusparse-dev-${CUDA_MAJOR_VERSION}-${CUDA_MINOR_VERSION} \
@@ -139,6 +140,7 @@ RUN if [ "${BUILD_TYPE}" = "cublas" ]; then \
139
140
apt-get update && \
140
141
apt-get install -y --no-install-recommends \
141
142
cuda-nvcc-${CUDA_MAJOR_VERSION}-${CUDA_MINOR_VERSION} \
143
+ libcufft-dev-${CUDA_MAJOR_VERSION}-${CUDA_MINOR_VERSION} \
142
144
libcurand-dev-${CUDA_MAJOR_VERSION}-${CUDA_MINOR_VERSION} \
143
145
libcublas-dev-${CUDA_MAJOR_VERSION}-${CUDA_MINOR_VERSION} \
144
146
libcusparse-dev-${CUDA_MAJOR_VERSION}-${CUDA_MINOR_VERSION} \
Original file line number Diff line number Diff line change @@ -16,7 +16,7 @@ RWKV_REPO?=https://github.com/donomii/go-rwkv.cpp
16
16
RWKV_VERSION? =661e7ae26d442f5cfebd2a0881b44e8c55949ec6
17
17
18
18
# whisper.cpp version
19
- WHISPER_CPP_VERSION? =af5833e29819810f2d83228228a9a3077e5ccd93
19
+ WHISPER_CPP_VERSION? =ffef323c4cfa8596cb91cf92d6f791f01a44335e
20
20
21
21
# bert.cpp version
22
22
BERT_VERSION? =710044b124545415f555e4260d16b146c725a6e4
@@ -100,7 +100,7 @@ ifeq ($(BUILD_TYPE),cublas)
100
100
CGO_LDFLAGS+=-lcublas -lcudart -L$(CUDA_LIBPATH)
101
101
export LLAMA_CUBLAS=1
102
102
export WHISPER_CUDA=1
103
- CGO_LDFLAGS_WHISPER+=-L$(CUDA_LIBPATH)/stubs/ -lcuda
103
+ CGO_LDFLAGS_WHISPER+=-L$(CUDA_LIBPATH)/stubs/ -lcuda -lcufft
104
104
endif
105
105
106
106
ifeq ($(BUILD_TYPE ) ,hipblas)
You can’t perform that action at this time.
0 commit comments