Stanley Tsang
73 строки
1.7 KiB
Makefile
73 строки
1.7 KiB
Makefile
#
|
|
# Copyright (c) 2015-2019, NVIDIA CORPORATION. All rights reserved.
|
|
# Modifications are Copyright (c) 2019 Advanced Micro Devices, Inc. All rights reserved.
|
|
#
|
|
# See LICENSE.txt for license information
|
|
#
|
|
|
|
ROCM_HOME ?= /opt/rocm
|
|
MPI_HOME ?= /usr/lib/openmpi
|
|
PREFIX ?= /usr/local
|
|
VERBOSE ?= 0
|
|
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
|
|
|
|
ifeq ($(DEBUG), 0)
|
|
HIPCUFLAGS += -O3
|
|
else
|
|
HIPCUFLAGS += -O0 -g -ggdb3
|
|
endif
|
|
|
|
ifeq ($(VERBOSE), 0)
|
|
.SILENT:
|
|
endif
|
|
|
|
.PHONY: build clean
|
|
|
|
BUILDDIR ?= ../build
|
|
ifneq ($(NCCL_HOME), "")
|
|
HIPCUFLAGS += -I$(NCCL_HOME)/include/
|
|
HIPLDFLAGS += -L$(NCCL_HOME)/lib
|
|
endif
|
|
|
|
ifeq ($(MPI), 1)
|
|
HIPCUFLAGS += -DMPI_SUPPORT -I${MPI_HOME}/include
|
|
HIPLDFLAGS += -L${MPI_HOME}/lib -lmpi
|
|
endif
|
|
LIBRARIES += rccl
|
|
HIPLDFLAGS += $(LIBRARIES:%=-l%)
|
|
|
|
DST_DIR := $(BUILDDIR)
|
|
SRC_FILES := $(wildcard *.cu)
|
|
OBJ_FILES := $(SRC_FILES:%.cu=${DST_DIR}/%.o)
|
|
BIN_FILES_LIST := all_reduce all_gather broadcast reduce_scatter reduce
|
|
BIN_FILES := $(BIN_FILES_LIST:%=${DST_DIR}/%_perf)
|
|
|
|
build: ${BIN_FILES}
|
|
|
|
clean:
|
|
rm -rf ${DST_DIR}
|
|
|
|
${DST_DIR}/%.o: %.cu common.h
|
|
@printf "Compiling %-35s > %s\n" $< $@
|
|
@mkdir -p ${DST_DIR}
|
|
echo "$(HIPCC) -o $@ $(HIPCUFLAGS) -c $<"
|
|
$(HIPCC) -o $@ $(HIPCUFLAGS) -c $<
|
|
|
|
${DST_DIR}/%_perf:${DST_DIR}/%.o ${DST_DIR}/common.o
|
|
@printf "Linking %-35s > %s\n" $< $@
|
|
@mkdir -p ${DST_DIR}
|
|
echo "$(HIPCC) -o $@ $(HIPCUFLAGS) $^ ${HIPLDFLAGS}"
|
|
$(HIPCC) -o $@ $(HIPCUFLAGS) $^ ${HIPLDFLAGS}
|
|
|