Skip to content

Commit cfbf7bd

Browse files
committed
fix: add LLAMA_CUBLAS on BUILD_TYPE=cublas
Thanks to @Thireus for noticing it. See: #258
1 parent aa7a18f commit cfbf7bd

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

Makefile

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,6 @@ BUILD_TYPE?=
1616
CGO_LDFLAGS?=
1717
CUDA_LIBPATH?=/usr/local/cuda/lib64/
1818
STABLEDIFFUSION_VERSION?=c0748eca3642d58bcf9521108bcee46959c647dc
19-
2019
GO_TAGS?=
2120

2221
OPTIONAL_TARGETS?=
@@ -36,9 +35,9 @@ endif
3635

3736
ifeq ($(BUILD_TYPE),cublas)
3837
CGO_LDFLAGS+=-lcublas -lcudart -L$(CUDA_LIBPATH)
38+
LLAMA_CUBLAS=1
3939
endif
4040

41-
4241
ifeq ($(GO_TAGS),stablediffusion)
4342
OPTIONAL_TARGETS+=go-stable-diffusion/libstablediffusion.a
4443
endif

0 commit comments

Comments
 (0)