fix(whisper.cpp): Add stubs and -lcuda

This commit is contained in:
Ettore Di Giacinto committed 2024-03-18 12:13:39 +01:00
1 parent 0663f66205
commit b790fca180
1 file changed
+3
+3
View File
@@ -91,10 +91,13 @@ ifeq ($(BUILD_TYPE),openblas)
export WHISPER_OPENBLAS=1
endif
ifeq ($(BUILD_TYPE),cublas)
CGO_LDFLAGS+=-lcublas -lcudart -lculibos -lcublasLt -L$(CUDA_LIBPATH)
export LLAMA_CUBLAS=1
# required by whisper.cpp
export WHISPER_CUBLAS=1
CGO_LDFLAGS+=-L$(CUDA_PATH)/stubs -lcuda
endif
ifeq ($(BUILD_TYPE),hipblas)