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

This commit is contained in:
Gilbert Lee
2019-04-22 10:25:49 -07:00
förälder 4474fe168d
incheckning 10e1572f72
+2 -1
Visa fil
@@ -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