From 08a59332af23a7d9803e09e712bea077bd8d571c Mon Sep 17 00:00:00 2001 From: Gilbert Lee Date: Mon, 22 Apr 2019 10:25:49 -0700 Subject: [PATCH] Adding way to specify a custom RCCL shared library file to link against [ROCm/rccl-tests commit: 10e1572f726054ccef30f08526897e5f08fbe54c] --- projects/rccl-tests/src/Makefile | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/projects/rccl-tests/src/Makefile b/projects/rccl-tests/src/Makefile index 56d52405bb..f1a6a6e2e3 100644 --- a/projects/rccl-tests/src/Makefile +++ b/projects/rccl-tests/src/Makefile @@ -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