Skip to content

Commit 60c0fe8

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

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

Makefile

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@ BUILD_TYPE?=
1616
CGO_LDFLAGS?=
1717
CUDA_LIBPATH?=/usr/local/cuda/lib64/
1818
STABLEDIFFUSION_VERSION?=c0748eca3642d58bcf9521108bcee46959c647dc
19-
19+
LLAMA_CUBLAS=0
2020
GO_TAGS?=
2121

2222
OPTIONAL_TARGETS?=
@@ -36,9 +36,9 @@ endif
3636

3737
ifeq ($(BUILD_TYPE),cublas)
3838
CGO_LDFLAGS+=-lcublas -lcudart -L$(CUDA_LIBPATH)
39+
LLAMA_CUBLAS=1
3940
endif
4041

41-
4242
ifeq ($(GO_TAGS),stablediffusion)
4343
OPTIONAL_TARGETS+=go-stable-diffusion/libstablediffusion.a
4444
endif
@@ -147,7 +147,7 @@ go-llama:
147147
cd go-llama && git checkout -b build $(GOLLAMA_VERSION) && git submodule update --init --recursive --depth 1
148148

149149
go-llama/libbinding.a: go-llama
150-
$(MAKE) -C go-llama BUILD_TYPE=$(BUILD_TYPE) libbinding.a
150+
LLAMA_CUBLAS=$(LLAMA_CUBLAS) $(MAKE) -C go-llama BUILD_TYPE=$(BUILD_TYPE) libbinding.a
151151

152152
replace:
153153
$(GOCMD) mod edit -replace github.com/go-skynet/go-llama.cpp=$(shell pwd)/go-llama

0 commit comments

Comments
 (0)