From bf3d936aea5b22619e86fab423d0f7170abf4206 Mon Sep 17 00:00:00 2001 From: Ettore Di Giacinto Date: Fri, 19 May 2023 17:11:28 +0200 Subject: [PATCH] fix: add LLAMA_CUBLAS on BUILD_TYPE=cublas (#310) --- Makefile | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/Makefile b/Makefile index bee2a91..35ab79a 100644 --- a/Makefile +++ b/Makefile @@ -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