gregandev
/
FlaskAI
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

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

Browse Source
swagger2
Ettore Di Giacinto 1 year ago committed by GitHub
parent 19deea986a
commit bf3d936aea
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 2 deletions
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Show Stats Download Patch File Download Diff File
  1. 3
      Makefile

3
Makefile
Unescape View File

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

Write Preview
Loading…
Cancel
Save