Build fastsocket plugin from ext-net
This commit is contained in:
parent
c88c9f873f
commit
c5790b3672
22
ext-net/google-fastsocket/Makefile
Normal file
22
ext-net/google-fastsocket/Makefile
Normal file
@ -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
|
Loading…
x
Reference in New Issue
Block a user