Adding way to specify a custom RCCL shared library file to link against

[ROCm/rccl-tests commit: 10e1572f72]
This commit is contained in:
Gilbert Lee
2019-04-22 10:25:49 -07:00
parent 8ff34620fb
commit 08a59332af
+2 -1
Vedi File
@@ -14,13 +14,14 @@ DEBUG ?= 0
HIPCC = $(ROCM_HOME)/hip/bin/hipcc
CXX = $(HIPCC)
HIPCUFLAGS :=
HIPCUFLAGS += -I$(ROCM_HOME)/include
HIPCUFLAGS += -I$(ROCM_HOME)/include/rccl
HIPCUFLAGS += -I$(ROCM_HOME)/hip/include/hip
HIPCUFLAGS += -I$(ROCM_HOME)/hiprand/include
LDFLAGS := -L$(ROCM_HOME)/lib -lhsa-runtime64 -lrt
HIPLDFLAGS := -L$(ROCM_HOME)/lib -lhsa-runtime64 -lrt
HIPLDFLAGS := $(CUSTOM_RCCL_LIB) -L$(ROCM_HOME)/lib -lhsa-runtime64 -lrt
ifeq ($(DEBUG), 0)
HIPCUFLAGS += -O3