allow adapting wormhole

rabit-learn/make/common.mk

@@ -3,6 +3,14 @@
 export LDFLAGS= -L../../lib -pthread -lm -lrt
 export CFLAGS = -Wall  -msse2  -Wno-unknown-pragmas -fPIC -I../../include  
 
+# setup opencv
+ifeq ($(USE_WORMHOLE),1)
+	CFLAGS+= -DRABIT_USE_WORMHOLE=1 -I ../../wormhole/include
+	LDFLAGS+= -L../../wormhole -lwormhole
+else
+	CFLAGS+= -DRABIT_USE_WORMHOLE=0
+endif
+
 # setup opencv
 ifeq ($(USE_HDFS),1)
 	CFLAGS+= -DRABIT_USE_HDFS=1 -I$(HADOOP_HDFS_HOME)/include -I$(JAVA_HOME)/include
@@ -11,6 +19,7 @@ else
 	CFLAGS+= -DRABIT_USE_HDFS=0
 endif
 
+
 .PHONY: clean all lib mpi
 
 all: $(BIN) $(MOCKBIN)

rabit-learn/make/config.mk

@@ -17,5 +17,8 @@ export MPICXX = mpicxx
 # whether use HDFS support during compile
 USE_HDFS = 1
 
+# whether use wormhole's io utils
+USE_WORMHOLE = 0
+
 # path to libjvm.so
 LIBJVM=$(JAVA_HOME)/jre/lib/amd64/server