aboutsummaryrefslogtreecommitdiff
path: root/Makefile
diff options
context:
space:
mode:
authorJohannes Gäßler <johannesg@5d6.de>2023-04-28 15:40:32 +0200
committerGitHub <noreply@github.com>2023-04-28 15:40:32 +0200
commit92a6e13a31ba052abd9062af6cb8df2a293ce661 (patch)
treed53852cd95e4d8bcf2644e7d2f92eca751be218a /Makefile
parent04aaae1d79482cad2564412f3b32e70298ac7789 (diff)
Add Manjaro CUDA include and lib dirs to Makefile (#1212)
Diffstat (limited to 'Makefile')
-rw-r--r--Makefile4
1 files changed, 2 insertions, 2 deletions
diff --git a/Makefile b/Makefile
index 8fbb19c..f7c8dbf 100644
--- a/Makefile
+++ b/Makefile
@@ -105,8 +105,8 @@ ifdef LLAMA_OPENBLAS
LDFLAGS += -lopenblas
endif
ifdef LLAMA_CUBLAS
- CFLAGS += -DGGML_USE_CUBLAS -I/usr/local/cuda/include
- LDFLAGS += -lcublas -lculibos -lcudart -lcublasLt -lpthread -ldl -lrt -L/usr/local/cuda/lib64
+ CFLAGS += -DGGML_USE_CUBLAS -I/usr/local/cuda/include -I$(CUDA_PATH)/targets/x86_64-linux/include
+ LDFLAGS += -lcublas -lculibos -lcudart -lcublasLt -lpthread -ldl -lrt -L/usr/local/cuda/lib64 -L$(CUDA_PATH)/targets/x86_64-linux/lib
OBJS += ggml-cuda.o
NVCC = nvcc
NVCCFLAGS = --forward-unknown-to-host-compiler -arch=native