diff --git a/ext-net/google-fastsocket/Makefile b/ext-net/google-fastsocket/Makefile new file mode 100644 index 0000000..e40e305 --- /dev/null +++ b/ext-net/google-fastsocket/Makefile @@ -0,0 +1,22 @@ +CUDA_HOME?=/usr/local/cuda +INC:=-I$(CUDA_HOME)/include +PLUGIN_SO:=../../build/libnccl-net.so + +default: $(PLUGIN_SO) + +$(PLUGIN_SO): nccl-fastsocket/net_fastsocket.cc nccl-fastsocket/compat.cc + $(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