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

이 커밋은 다음에 포함됨:
Gilbert Lee가 2019-04-22 10:25:49 -07:00를 커밋했습니다
1 부모 4474fe168d
커밋 10e1572f72
1 파일이 변경됨
+2 -1
+2 -1
파일 보기
@@ -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