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

1 year ago · bf3d936aea
parent 19deea986a
commit bf3d936aea
1 changed files with 1 additions and 2 deletions
--- a/3
+++ b/3
@ -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