fix: add LLAMA_CUBLAS on BUILD_TYPE=cublas (#310)

This commit is contained in:
Ettore Di Giacinto
2023-05-19 17:11:28 +02:00
committed by GitHub
parent 19deea986a
commit bf3d936aea

View File

@@ -16,7 +16,6 @@ BUILD_TYPE?=
CGO_LDFLAGS?=
CUDA_LIBPATH?=/usr/local/cuda/lib64/
STABLEDIFFUSION_VERSION?=c0748eca3642d58bcf9521108bcee46959c647dc
GO_TAGS?=
OPTIONAL_TARGETS?=
@@ -36,9 +35,9 @@ endif
ifeq ($(BUILD_TYPE),cublas)
CGO_LDFLAGS+=-lcublas -lcudart -L$(CUDA_LIBPATH)
export LLAMA_CUBLAS=1
endif
ifeq ($(GO_TAGS),stablediffusion)
OPTIONAL_TARGETS+=go-stable-diffusion/libstablediffusion.a
endif