5797dcb64e
* support run rabit tests as xgboost subproject using xgboost/dmlc-core * support tracker config set/get * remove redudant printf * remove redudant printf * add c++0x declaration * log allreduce/broadcast caller, engine should track caller stack for investigation * tracker support binary config format * Revert "tracker support binary config format" This reverts commit 2a28e5e2b55c200cb621af8d19f17ab1bc62503b. * remove caller, prototype fetch allreduce/broadcast results from resbuf * store cached allreduce/broadcast seq_no to tracker * allow restore all caches from other nodes * try new rabit collective cache, todo: recv_link seems down * link up cache restore with main recovery * cleanup load cache state * update cache api * pass test.mk * have a working tests * try to unify check into actionsummary * more logging to debug distributed hist three method issue * update rabit interface to support caller signature matching * splite seq_counter from cur_cache_seq to different variables * still see issue with inf loop * support debug print caller as well as allreduce op * cleanup * remove get/set cache from model_recover, adding recover in loadcheckpoint * clarify rabit cache strategy, cache is set only by successful collective call involving all nodes with unique cache key. if all nodes call getcache at same time, we keep rabit run collective call. If some nodes call getcache while others not, we backfill cache from those nodes with most entries * revert caller logs * fix lint error * fix engine mpi signature * support getcache by ref * allow result buffer presiet to filestream * add loging * try fix checkpoint failure recovery case * use int64_t to avoid overflow caused seq fault * try avoid int overflow * try fix checkpoint failure recovery case * try avoid seqno overflow to negative by offseting specifial flag value adding cache seq no to checkpoint/load checkpoint/check point ack to avoid confusion from cache recovery * fix cache seq assert error * remove loging, handle edge case * add extensive log to checkpoint state with different seq no * fix lint errors * clean up comments before merge back to master * add logs to allreduce/broadcast/checkpoint * use unsinged int 32 and give seq no larger range * address remove allreduce dropseq code segment * using caller signature to filter bootstrapallreduces * remove get/set cache from empty * apply signature to reducer * apply signature to broadcast * add key to broadcat log * fix broadcast signature * fix default _line value for non linux system * adding comments, remove sleep(1) * fix osx build issue * try fix mpi * fix doc * fix engine_empty api * logging, adding more logs, restore immutable assertion * print unsinged int with ud * fix lint * rename seqtype to kSeq and KCache indicating it's usage apply kDiffSeq check to load_cache routine * comment allreduce/broadcast log * allow tests run on arm * enable flag to turn on / off cache * add log info alert if user choose to enable rabit bootstrap cache * add rabit_debug setting so user can use config to turn on * log flags when user turn on rabit_debug * force rabit restart if tracker assign -1 rank * use OPENMP to vecotrize reducer * address comment * Revert "address comment" This reverts commit 1dc61f33e7357dad8fa65528abeb81db92c5f9ed. * fix checkpoint size print 0 * per feedback, remove DISABLEOPEMP, address race condition * - remove openmp from this pr - update name from cache to boostrapcache * add default value of signature macros * remove openmp from cmake file * Update src/allreduce_robust.cc Co-Authored-By: Philip Hyunsu Cho <chohyu01@cs.washington.edu> * Update src/allreduce_robust.cc Co-Authored-By: Philip Hyunsu Cho <chohyu01@cs.washington.edu> * run test with cmake * remove openmp * fix cmake based tests * use cmake test fix darwin .dylib issue * move around rabit_signature definition due to windows build * misc, add c++ check in CMakeFile * per feedback * resolve CMake file * update rabit version
105 lines
3.0 KiB
Makefile
105 lines
3.0 KiB
Makefile
OS := $(shell uname)
|
|
|
|
RABIT_BUILD_DMLC = 0
|
|
|
|
export WARNFLAGS= -Wall -Wextra -Wno-unused-parameter -Wno-unknown-pragmas -std=c++11
|
|
export CFLAGS = -O3 $(WARNFLAGS)
|
|
export LDFLAGS =-Llib
|
|
|
|
#download mpi
|
|
#echo $(shell scripts/mpi.sh)
|
|
|
|
MPICXX=./mpich/bin/mpicxx
|
|
|
|
export CXX = g++
|
|
|
|
|
|
#----------------------------
|
|
# Settings for power and arm arch
|
|
#----------------------------
|
|
ARCH := $(shell uname -a)
|
|
ifneq (,$(filter $(ARCH), armv6l armv7l powerpc64le ppc64le aarch64))
|
|
CFLAGS += -march=native
|
|
else
|
|
CFLAGS += -msse2
|
|
endif
|
|
|
|
ifndef WITH_FPIC
|
|
WITH_FPIC = 1
|
|
endif
|
|
ifeq ($(WITH_FPIC), 1)
|
|
CFLAGS += -fPIC
|
|
endif
|
|
|
|
ifndef LINT_LANG
|
|
LINT_LANG="all"
|
|
endif
|
|
|
|
ifeq ($(RABIT_BUILD_DMLC),1)
|
|
DMLC=dmlc-core
|
|
else
|
|
DMLC=../dmlc-core
|
|
endif
|
|
|
|
CFLAGS += -I $(DMLC)/include -I include/
|
|
|
|
# build path
|
|
BPATH=.
|
|
# objectives that makes up rabit library
|
|
MPIOBJ= $(BPATH)/engine_mpi.o
|
|
OBJ= $(BPATH)/allreduce_base.o $(BPATH)/allreduce_robust.o $(BPATH)/engine.o $(BPATH)/engine_empty.o $(BPATH)/engine_mock.o\
|
|
$(BPATH)/c_api.o $(BPATH)/engine_base.o
|
|
SLIB= lib/librabit.so lib/librabit_mock.so lib/librabit_base.so
|
|
ALIB= lib/librabit.a lib/librabit_empty.a lib/librabit_mock.a lib/librabit_base.a
|
|
MPISLIB= lib/librabit_mpi.so
|
|
MPIALIB= lib/librabit_mpi.a
|
|
HEADERS=src/*.h include/rabit/*.h include/rabit/internal/*.h
|
|
|
|
.PHONY: clean all install mpi python lint doc doxygen
|
|
|
|
all: lib/librabit.a lib/librabit_mock.a lib/librabit.so lib/librabit_base.a lib/librabit_mock.so
|
|
mpi: lib/librabit_mpi.a lib/librabit_mpi.so
|
|
|
|
$(BPATH)/allreduce_base.o: src/allreduce_base.cc $(HEADERS)
|
|
$(BPATH)/engine.o: src/engine.cc $(HEADERS)
|
|
$(BPATH)/allreduce_robust.o: src/allreduce_robust.cc $(HEADERS)
|
|
$(BPATH)/engine_mpi.o: src/engine_mpi.cc $(HEADERS)
|
|
$(BPATH)/engine_empty.o: src/engine_empty.cc $(HEADERS)
|
|
$(BPATH)/engine_mock.o: src/engine_mock.cc $(HEADERS)
|
|
$(BPATH)/engine_base.o: src/engine_base.cc $(HEADERS)
|
|
$(BPATH)/c_api.o: src/c_api.cc $(HEADERS)
|
|
|
|
lib/librabit.a lib/librabit.so: $(BPATH)/allreduce_base.o $(BPATH)/allreduce_robust.o $(BPATH)/engine.o $(BPATH)/c_api.o
|
|
lib/librabit_base.a lib/librabit_base.so: $(BPATH)/allreduce_base.o $(BPATH)/engine_base.o $(BPATH)/c_api.o
|
|
lib/librabit_mock.a lib/librabit_mock.so: $(BPATH)/allreduce_base.o $(BPATH)/allreduce_robust.o $(BPATH)/engine_mock.o $(BPATH)/c_api.o
|
|
lib/librabit_empty.a: $(BPATH)/engine_empty.o $(BPATH)/c_api.o
|
|
lib/librabit_mpi.a lib/librabit_mpi.so: $(MPIOBJ)
|
|
|
|
$(OBJ) :
|
|
$(CXX) -c $(CFLAGS) -o $@ $(firstword $(filter %.cpp %.c %.cc, $^) )
|
|
|
|
$(ALIB):
|
|
ar cr $@ $+
|
|
|
|
$(SLIB) :
|
|
$(CXX) $(CFLAGS) -shared -o $@ $(filter %.cpp %.o %.c %.cc %.a, $^) $(LDFLAGS)
|
|
|
|
$(MPIOBJ) :
|
|
$(MPICXX) -c $(CFLAGS) -I./mpich/include -o $@ $(firstword $(filter %.cpp %.c %.cc, $^) )
|
|
|
|
$(MPIALIB):
|
|
ar cr $@ $+
|
|
|
|
$(MPISLIB) :
|
|
$(MPICXX) $(CFLAGS) -I./mpich/include -shared -o $@ $(filter %.cpp %.o %.c %.cc %.a, $^) \
|
|
$(LDFLAGS) -L./mpich/lib -Wl,-rpath,./mpich/lib -lmpi
|
|
|
|
lint:
|
|
$(DMLC)/scripts/lint.py rabit $(LINT_LANG) src include
|
|
|
|
doc doxygen:
|
|
cd include; doxygen ../doc/Doxyfile; cd -
|
|
|
|
clean:
|
|
$(RM) $(OBJ) $(MPIOBJ) $(ALIB) $(MPIALIB) $(SLIB) *~ src/*~ include/*~ include/*/*~
|