2020-11-10 15:44:10 -07: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)
|
2020-11-10 15:44:10 -07:00
|
|
|
ifeq (,$(HIP_PATH))
|
|
|
|
|
HIP_PATH=../../..
|
|
|
|
|
endif
|
|
|
|
|
HIPCC=$(HIP_PATH)/bin/hipcc
|
|
|
|
|
|
|
|
|
|
EXE=HelloRccl
|
|
|
|
|
CXXFLAGS = -std=c++11 -O3 -I../../src/include -I$(RCCL_INSTALL) -L$(RCCL_INSTALL) -lrccl
|
|
|
|
|
|
|
|
|
|
all: $(EXE)
|
|
|
|
|
|
|
|
|
|
$(EXE): $(EXE).cpp $(shell find -regex ".*\.\hpp")
|
|
|
|
|
$(HIPCC) $(CXXFLAGS) $< -o $@
|
|
|
|
|
|
|
|
|
|
clean:
|
|
|
|
|
rm -f *.o $(EXE)
|
