mirror of
https://github.com/microsoft/mscclpp.git
synced 2026-05-02 04:31:28 +00:00
c580e4c503
* Allow CudaIpc connection between GPUs in a single process * Added an example of connection in a single process * Minor interface updates --------- Co-authored-by: Binyang Li <binyli@microsoft.com>
23 lines
466 B
Makefile
23 lines
466 B
Makefile
CUDA_HOME ?= /usr/local/cuda
|
|
ROCM_HOME ?= /opt/rocm
|
|
|
|
# Check if nvcc exists, otherwise use hipcc
|
|
ifeq ($(shell which $(CUDA_HOME)/bin/nvcc 2>/dev/null),)
|
|
COMPILER := $(ROCM_HOME)/bin/hipcc
|
|
ARCH_FLAG := -D__HIP_PLATFORM_AMD__=1
|
|
else
|
|
COMPILER := $(CUDA_HOME)/bin/nvcc
|
|
ARCH_FLAG := -arch=native
|
|
endif
|
|
|
|
TARGET = gpu_ping_pong
|
|
SRC = gpu_ping_pong.cu
|
|
|
|
all: $(TARGET)
|
|
|
|
$(TARGET): $(SRC)
|
|
$(COMPILER) $(ARCH_FLAG) -o $@ $< -lmscclpp
|
|
|
|
clean:
|
|
rm -f $(TARGET)
|