|
@@ -102,7 +102,7 @@ ifdef LLAMA_OPENBLAS
|
|
|
endif
|
|
endif
|
|
|
ifdef LLAMA_CUBLAS
|
|
ifdef LLAMA_CUBLAS
|
|
|
CFLAGS += -DGGML_USE_CUBLAS -I/usr/local/cuda/include
|
|
CFLAGS += -DGGML_USE_CUBLAS -I/usr/local/cuda/include
|
|
|
- LDFLAGS += -lcublas_static -lculibos -lcudart_static -lcublasLt_static -lpthread -ldl -L/usr/local/cuda/lib64
|
|
|
|
|
|
|
+ LDFLAGS += -lcublas_static -lculibos -lcudart_static -lcublasLt_static -lpthread -ldl -lrt -L/usr/local/cuda/lib64
|
|
|
OBJS += ggml-cuda.o
|
|
OBJS += ggml-cuda.o
|
|
|
ggml-cuda.o: ggml-cuda.cu ggml-cuda.h
|
|
ggml-cuda.o: ggml-cuda.cu ggml-cuda.h
|
|
|
nvcc -arch=native -c -o $@ $<
|
|
nvcc -arch=native -c -o $@ $<
|
源文雨