2024-05-13 10:56:32 -06:00
|
|
|
ROCM_DIR ?= /opt/rocm
|
2023-07-24 16:26:22 -06:00
|
|
|
RCCL_DIR ?= ../../build/release
|
2024-05-13 10:56:32 -06:00
|
|
|
MPI_DIR ?= /opt/ompi
|
|
|
|
|
MPI_INC_DIR ?= /usr/include/x86_64-linux-gnu/mpi
|
|
|
|
|
MPI_LIB_DIR ?= /usr/lib/x86_64-linux-gnu
|
|
|
|
|
|
2024-10-22 10:41:08 -04:00
|
|
|
INCLUDES = -I$(MPI_INC_DIR) -I$(MPI_DIR)/include -I$(RCCL_DIR)/include -I$(RCCL_DIR)/hipify/src/include
|
2024-05-13 10:56:32 -06:00
|
|
|
LDFLAGS = -L$(MPI_LIB_DIR) -L$(MPI_DIR)/lib -L$(RCCL_DIR) -lmpi -lrccl
|
2023-07-24 16:26:22 -06:00
|
|
|
|
|
|
|
|
main: rcclReplayer.cpp
|
2024-05-13 10:56:32 -06:00
|
|
|
$(ROCM_DIR)/bin/hipcc rcclReplayer.cpp -O1 -g -o rcclReplayer $(INCLUDES) $(LDFLAGS)
|
2023-07-24 16:26:22 -06:00
|
|
|
|
|
|
|
|
clean:
|
2024-05-13 10:56:32 -06:00
|
|
|
rm -f ./rcclReplayer
|
