Use ROCM_PATH instead of ROCM_HOME

This commit is contained in:
Wenkai Du
2021-07-21 14:19:48 -07:00
parent 2d9be62621
commit cc34c54509
+7 -7
Vedi File
@@ -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