From ef13aaf37926c30ca0b18650570b029d1d475805 Mon Sep 17 00:00:00 2001 From: Ubuntu Date: Sat, 11 Apr 2015 05:29:07 +0000 Subject: [PATCH] ch --- rabit-learn/io/io-inl.h | 6 +++--- rabit-learn/io/io.h | 8 ++++---- 2 files changed, 7 insertions(+), 7 deletions(-) diff --git a/rabit-learn/io/io-inl.h b/rabit-learn/io/io-inl.h index 95897ac09..fc7faae2e 100644 --- a/rabit-learn/io/io-inl.h +++ b/rabit-learn/io/io-inl.h @@ -10,7 +10,7 @@ #include "./io.h" -#if RABIT_USE_WORMHOLE == 0 +#if RABIT_USE_DMLC == 0 #if RABIT_USE_HDFS #include "./hdfs-inl.h" #endif @@ -28,7 +28,7 @@ namespace io { inline InputSplit *CreateInputSplit(const char *uri, unsigned part, unsigned nsplit) { -#if RABIT_USE_WORMHOLE +#if RABIT_USE_DMLC return dmlc::InputSplit::Create(uri, part, nsplit); #else using namespace std; @@ -76,7 +76,7 @@ class StreamAdapter : public Stream { * \param mode can be 'w' or 'r' for read or write */ inline Stream *CreateStream(const char *uri, const char *mode) { -#if RABIT_USE_WORMHOLE +#if RABIT_USE_DMLC return new StreamAdapter(dmlc::Stream::Create(uri, mode)); #else using namespace std; diff --git a/rabit-learn/io/io.h b/rabit-learn/io/io.h index dd766bd6d..1dd4d3d68 100644 --- a/rabit-learn/io/io.h +++ b/rabit-learn/io/io.h @@ -13,11 +13,11 @@ #define RABIT_USE_HDFS 0 #endif -#ifndef RABIT_USE_WORMHOLE -#define RABIT_USE_WORMHOLE 0 +#ifndef RABIT_USE_DMLC +#define RABIT_USE_DMLC 0 #endif -#if RABIT_USE_WORMHOLE +#if RABIT_USE_DMLC #include #endif /*! \brief io interface */ @@ -27,7 +27,7 @@ namespace rabit { */ namespace io { /*! \brief reused SeekStream's definition */ -#if RABIT_USE_WORMHOLE +#if RABIT_USE_DMLC typedef dmlc::SeekStream SeekStream; typedef dmlc::InputSplit InputSplit; #else