2022-03-14 14:55:24 -04:00
|
|
|
# Copyright (c) 2020 Advanced Micro Devices, Inc. All rights reserved.
|
|
|
|
|
|
|
|
|
|
# Set to where RCCL is installed
|
|
|
|
|
RCCL_INSTALL=../../build/release
|
|
|
|
|
|
2022-08-04 13:38:41 -04:00
|
|
|
HIP_PATH?= $(wildcard /opt/rocm)
|
2022-03-14 14:55:24 -04:00
|
|
|
ifeq (,$(HIP_PATH))
|
|
|
|
|
HIP_PATH=../../..
|
|
|
|
|
endif
|
|
|
|
|
HIPCC=$(HIP_PATH)/bin/hipcc
|
|
|
|
|
|
|
|
|
|
EXE=rccl-allreduce-multirank rccl-reducescatter-multirank
|
|
|
|
|
CXXFLAGS = -std=c++11 -O3 -I$(RCCL_INSTALL)/include/rccl/ -L$(RCCL_INSTALL) -lrccl
|
|
|
|
|
|
|
|
|
|
all: $(EXE)
|
|
|
|
|
|
|
|
|
|
rccl-allreduce-multirank: rccl-allreduce-multirank.cc $(shell find -regex ".*\.\hpp")
|
|
|
|
|
$(HIPCC) $(CXXFLAGS) $< -o $@
|
|
|
|
|
|
|
|
|
|
rccl-reducescatter-multirank: rccl-reducescatter-multirank.cc $(shell find -regex ".*\.\hpp")
|
|
|
|
|
$(HIPCC) $(CXXFLAGS) $< -o $@
|
|
|
|
|
|
|
|
|
|
clean:
|
|
|
|
|
rm -f *.o $(EXE)
|
