Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

ch

Browse Source
This commit is contained in:
Ubuntu 2015-04-11 05:29:07 +00:00
parent 50a66b3855
commit ef13aaf379
2 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
rabit-learn/io/io-inl.h
View File

@ -10,7 +10,7 @@
#include "./io.h" #include "./io.h"
#if RABIT_USE_WORMHOLE == 0 #if RABIT_USE_DMLC == 0
#if RABIT_USE_HDFS #if RABIT_USE_HDFS
#include "./hdfs-inl.h" #include "./hdfs-inl.h"
#endif #endif
@ -28,7 +28,7 @@ namespace io {
inline InputSplit *CreateInputSplit(const char *uri, inline InputSplit *CreateInputSplit(const char *uri,
unsigned part, unsigned part,
unsigned nsplit) { unsigned nsplit) {
#if RABIT_USE_WORMHOLE #if RABIT_USE_DMLC
return dmlc::InputSplit::Create(uri, part, nsplit); return dmlc::InputSplit::Create(uri, part, nsplit);
#else #else
using namespace std; using namespace std;
@ -76,7 +76,7 @@ class StreamAdapter : public Stream {
* \param mode can be 'w' or 'r' for read or write * \param mode can be 'w' or 'r' for read or write
*/ */
inline Stream *CreateStream(const char *uri, const char *mode) { inline Stream *CreateStream(const char *uri, const char *mode) {
#if RABIT_USE_WORMHOLE #if RABIT_USE_DMLC
return new StreamAdapter<dmlc::Stream>(dmlc::Stream::Create(uri, mode)); return new StreamAdapter<dmlc::Stream>(dmlc::Stream::Create(uri, mode));
#else #else
using namespace std; using namespace std;

8
rabit-learn/io/io.h
View File

@ -13,11 +13,11 @@
#define RABIT_USE_HDFS 0 #define RABIT_USE_HDFS 0
#endif #endif
#ifndef RABIT_USE_WORMHOLE #ifndef RABIT_USE_DMLC
#define RABIT_USE_WORMHOLE 0 #define RABIT_USE_DMLC 0
#endif #endif
#if RABIT_USE_WORMHOLE #if RABIT_USE_DMLC
#include <dmlc/io.h> #include <dmlc/io.h>
#endif #endif
/*! \brief io interface */ /*! \brief io interface */
@ -27,7 +27,7 @@ namespace rabit {
*/ */
namespace io { namespace io {
/*! \brief reused SeekStream's definition */ /*! \brief reused SeekStream's definition */
#if RABIT_USE_WORMHOLE #if RABIT_USE_DMLC
typedef dmlc::SeekStream SeekStream; typedef dmlc::SeekStream SeekStream;
typedef dmlc::InputSplit InputSplit; typedef dmlc::InputSplit InputSplit;
#else #else