Fix: Issue with CUBLAS compilation error

Fix: Issue with CUBLAS compilation error
This commit is contained in:
B1gM8c 2023-04-22 21:19:02 +08:00
parent 2390573ce3
commit 4676a1d41b

View file

@ -109,7 +109,7 @@ ifdef LLAMA_CUBLAS
LDFLAGS += -lcublas -lculibos -lcudart -lcublasLt -lpthread -ldl -lrt -L/usr/local/cuda/lib64
OBJS += ggml-cuda.o
NVCC = nvcc
NVCCFLAGS = --forward-unknown-to-host-linker -arch=native --compiler-options -fPIC
NVCCFLAGS = --forward-unknown-to-host-compiler -arch=native
ggml-cuda.o: ggml-cuda.cu ggml-cuda.h
ifdef LLAMA_GPROF
CFLAGS += -pg