AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Use ROCM_PATH instead of ROCM_HOME

Browse Source
This commit is contained in:
Wenkai Du
2021-07-21 14:19:48 -07:00
parent 2d9be62621
commit cc34c54509
1 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/Makefile
+7 -7
View File
@@ -5,14 +5,14 @@
# See LICENSE.txt for license information # See LICENSE.txt for license information
# #
ROCM_HOME ?= /opt/rocm ROCM_PATH ?= /opt/rocm
MPI_HOME ?= /usr/lib/openmpi MPI_HOME ?= /usr/lib/openmpi
PREFIX ?= /usr/local PREFIX ?= /usr/local
VERBOSE ?= 0 VERBOSE ?= 0
DEBUG ?= 0 DEBUG ?= 0
NCCL_HOME ?= "" NCCL_HOME ?= ""
HIPCC = $(ROCM_HOME)/hip/bin/hipcc HIPCC = $(ROCM_PATH)/hip/bin/hipcc
CXX = $(HIPCC) CXX = $(HIPCC)
HIPCUFLAGS := -std=c++14 HIPCUFLAGS := -std=c++14
@@ -23,11 +23,11 @@ ifneq ($(NCCL_HOME), "")
HIPCUFLAGS += -I$(NCCL_HOME) HIPCUFLAGS += -I$(NCCL_HOME)
HIPLDFLAGS += -Wl,-rpath,$(NCCL_HOME) -L$(NCCL_HOME) HIPLDFLAGS += -Wl,-rpath,$(NCCL_HOME) -L$(NCCL_HOME)
endif endif
HIPCUFLAGS += -I$(ROCM_HOME)/include HIPCUFLAGS += -I$(ROCM_PATH)/include
HIPCUFLAGS += -I$(ROCM_HOME)/include/rccl HIPCUFLAGS += -I$(ROCM_PATH)/include/rccl
HIPCUFLAGS += -I$(ROCM_HOME)/hip/include/hip HIPCUFLAGS += -I$(ROCM_PATH)/hip/include/hip
LDFLAGS += -L$(ROCM_HOME)/lib -lhsa-runtime64 -lrt LDFLAGS += -L$(ROCM_PATH)/lib -lhsa-runtime64 -lrt
HIPLDFLAGS += $(CUSTOM_RCCL_LIB) -L$(ROCM_HOME)/lib -lhsa-runtime64 -lrt HIPLDFLAGS += $(CUSTOM_RCCL_LIB) -L$(ROCM_PATH)/lib -lhsa-runtime64 -lrt
ifeq ($(DEBUG), 0) ifeq ($(DEBUG), 0)
HIPCUFLAGS += -O3 HIPCUFLAGS += -O3