From cc34c545098145bc148e5035e4c8e767b4d71ece Mon Sep 17 00:00:00 2001 From: Wenkai Du Date: Wed, 21 Jul 2021 14:19:48 -0700 Subject: [PATCH] Use ROCM_PATH instead of ROCM_HOME --- src/Makefile | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/src/Makefile b/src/Makefile index 31e0fda431..cb2bdb09d5 100644 --- a/src/Makefile +++ b/src/Makefile @@ -5,14 +5,14 @@ # See LICENSE.txt for license information # -ROCM_HOME ?= /opt/rocm +ROCM_PATH ?= /opt/rocm MPI_HOME ?= /usr/lib/openmpi PREFIX ?= /usr/local VERBOSE ?= 0 DEBUG ?= 0 NCCL_HOME ?= "" -HIPCC = $(ROCM_HOME)/hip/bin/hipcc +HIPCC = $(ROCM_PATH)/hip/bin/hipcc CXX = $(HIPCC) HIPCUFLAGS := -std=c++14 @@ -23,11 +23,11 @@ ifneq ($(NCCL_HOME), "") HIPCUFLAGS += -I$(NCCL_HOME) HIPLDFLAGS += -Wl,-rpath,$(NCCL_HOME) -L$(NCCL_HOME) endif -HIPCUFLAGS += -I$(ROCM_HOME)/include -HIPCUFLAGS += -I$(ROCM_HOME)/include/rccl -HIPCUFLAGS += -I$(ROCM_HOME)/hip/include/hip -LDFLAGS += -L$(ROCM_HOME)/lib -lhsa-runtime64 -lrt -HIPLDFLAGS += $(CUSTOM_RCCL_LIB) -L$(ROCM_HOME)/lib -lhsa-runtime64 -lrt +HIPCUFLAGS += -I$(ROCM_PATH)/include +HIPCUFLAGS += -I$(ROCM_PATH)/include/rccl +HIPCUFLAGS += -I$(ROCM_PATH)/hip/include/hip +LDFLAGS += -L$(ROCM_PATH)/lib -lhsa-runtime64 -lrt +HIPLDFLAGS += $(CUSTOM_RCCL_LIB) -L$(ROCM_PATH)/lib -lhsa-runtime64 -lrt ifeq ($(DEBUG), 0) HIPCUFLAGS += -O3