Update Makefile

This commit is contained in:
FantasyGmm 2023-12-22 18:23:33 +08:00 committed by GitHub
parent 96f2634cf6
commit 2640a7434a
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -369,11 +369,10 @@ ifdef LLAMA_CUBLAS
MK_CPPFLAGS += -DGGML_USE_CUBLAS -I/usr/local/cuda/include -I/opt/cuda/include -I$(CUDA_PATH)/targets/x86_64-linux/include -I/usr/local/cuda/targets/aarch64-linux/include
MK_LDFLAGS += -lcublas -lculibos -lcudart -lcublasLt -lpthread -ldl -lrt -L/usr/local/cuda/lib64 -L/opt/cuda/lib64 -L$(CUDA_PATH)/targets/x86_64-linux/lib -L/usr/local/cuda/targets/aarch64-linux/lib
OBJS += ggml-cuda.o
ifdef JETSON_EOL_MODULE_DETECT
MK_NVCCFLAGS = -use_fast_math
else
MK_NVCCFLAGS = --forward-unknown-to-host-compiler -use_fast_math
endif # JETSON_EOL_MODULE_DETECT
ifndef JETSON_EOL_MODULE_DETECT
MK_NVCCFLAGS += --forward-unknown-to-host-compiler
endif # JETSON_EOL_MODULE_DETECT
ifdef LLAMA_DEBUG
MK_NVCCFLAGS += -lineinfo