File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -16,7 +16,7 @@ BUILD_TYPE?=
16
16
CGO_LDFLAGS? =
17
17
CUDA_LIBPATH? =/usr/local/cuda/lib64/
18
18
STABLEDIFFUSION_VERSION? =c0748eca3642d58bcf9521108bcee46959c647dc
19
-
19
+ LLAMA_CUBLAS =0
20
20
GO_TAGS? =
21
21
22
22
OPTIONAL_TARGETS? =
36
36
37
37
ifeq ($(BUILD_TYPE ) ,cublas)
38
38
CGO_LDFLAGS+=-lcublas -lcudart -L$(CUDA_LIBPATH)
39
+ LLAMA_CUBLAS=1
39
40
endif
40
41
41
-
42
42
ifeq ($(GO_TAGS ) ,stablediffusion)
43
43
OPTIONAL_TARGETS+=go-stable-diffusion/libstablediffusion.a
44
44
endif
@@ -147,7 +147,7 @@ go-llama:
147
147
cd go-llama && git checkout -b build $(GOLLAMA_VERSION ) && git submodule update --init --recursive --depth 1
148
148
149
149
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
151
151
152
152
replace :
153
153
$(GOCMD ) mod edit -replace github.com/go-skynet/go-llama.cpp=$(shell pwd) /go-llama
You can’t perform that action at this time.
0 commit comments