mirror of
https://github.com/microsoft/mscclpp.git
synced 2026-05-11 17:00:22 +00:00
2a6f1c1192
This PR adds an example code for switch channel testing. It validates switch channel on single node and multi node environments. We need to add the description of the algorithms and the explanation of the code under doc. example outputs: rank0: ./bidir_switch_channel 10.0.5.233:45571 0 0 Rank 0 (GPU 0): Preparing for tests ... Rank 0 (GPU 0): bytes 4096, elapsed 0.0062328 ms/iter, BW 0.657169 GB/s Rank 0 (GPU 0): bytes 4.1943e+06, elapsed 0.0164577 ms/iter, BW 254.854 GB/s Rank 0 (GPU 0): bytes 1.34218e+08, elapsed 0.33628 ms/iter, BW 399.125 GB/s Rank 0: Succeed! rank1: ./bidir_switch_channel 10.0.5.233:45571 1 0 Rank 1 (GPU 0): Preparing for tests ... Rank 1: Succeed!
16 lines
256 B
Makefile
16 lines
256 B
Makefile
CUDA_HOME ?= /usr/local/cuda
|
|
|
|
COMPILER := $(CUDA_HOME)/bin/nvcc
|
|
ARCH_FLAG := -arch=native
|
|
|
|
TARGET = bidir_switch_channel
|
|
SRC = bidir_switch_channel.cu
|
|
|
|
all: $(TARGET)
|
|
|
|
$(TARGET): $(SRC)
|
|
$(COMPILER) $(ARCH_FLAG) -o $@ $< -lmscclpp
|
|
|
|
clean:
|
|
rm -f $(TARGET)
|