add rpath

This commit is contained in:
Henri Vasserman 2023-04-21 03:32:06 +03:00
parent 0e005f7793
commit d3e1984ce0

View file

@ -109,7 +109,7 @@ ggml-cuda.o: ggml-cuda.cu ggml-cuda.h
endif
ifdef LLAMA_HIPBLAS
ROCM_PATH ?= /opt/rocm
LDFLAGS += -lhipblas -lamdhip64 -L$(ROCM_PATH)/lib
LDFLAGS += -lhipblas -lamdhip64 -L$(ROCM_PATH)/lib -Wl,-rpath=$(ROCM_PATH)/lib
HIPCC ?= $(ROCM_PATH)/bin/hipcc
OBJS += ggml-cuda.o
ggml.o: CFLAGS += -DGGML_USE_HIPBLAS -D__HIP_PLATFORM_AMD__ -I$(ROCM_PATH)/include