2021-12-08 11:43:36 -08:00
|
|
|
CUDA_HOME?=/usr/local/cuda
|
|
|
|
|
INC:=-I$(CUDA_HOME)/include
|
2022-01-24 07:19:48 -08:00
|
|
|
PLUGIN_SO:=libnccl-net.so
|
2021-12-08 11:43:36 -08:00
|
|
|
|
|
|
|
|
default: $(PLUGIN_SO)
|
|
|
|
|
|
2022-01-24 07:19:48 -08:00
|
|
|
$(PLUGIN_SO): nccl-fastsocket/net_fastsocket.cc nccl-fastsocket/compat.cc nccl-fastsocket/utilities.cc
|
2021-12-08 11:43:36 -08:00
|
|
|
$(CC) $(INC) -fPIC -shared -o $@ -Wl,-soname,$(PLUGIN_SO) $^
|
|
|
|
|
|
|
|
|
|
nccl-fastsocket/%.cc:
|
|
|
|
|
git clone https://github.com/google/nccl-fastsocket.git
|
|
|
|
|
|
|
|
|
|
install: $(BUILDDIR)/lib/$(PLUGIN_SO)
|
|
|
|
|
|
|
|
|
|
$(BUILDDIR)/lib/$(PLUGIN_SO): $(PLUGIN_SO)
|
|
|
|
|
@printf "Grabbing %-35s > %s\n" $< $@
|
|
|
|
|
mkdir -p $(BUILDDIR)/lib
|
|
|
|
|
install -m 644 $< $@
|
|
|
|
|
|
|
|
|
|
clean:
|
|
|
|
|
rm -f $(PLUGIN_SO)
|
|
|
|
|
rm -Rf nccl-fastsocket
|
