207f058711
* Refactor CMake scripts. * Remove CMake CUDA wrapper. * Bump CMake version for CUDA. * Use CMake to handle Doxygen. * Split up CMakeList. * Export install target. * Use modern CMake. * Remove build.sh * Workaround for gpu_hist test. * Use cmake 3.12. * Revert machine.conf. * Move CLI test to gpu. * Small cleanup. * Support using XGBoost as submodule. * Fix windows * Fix cpp tests on Windows * Remove duplicated find_package.
70 lines
1.8 KiB
C++
70 lines
1.8 KiB
C++
// Copyright (c) 2015 by Contributors
|
|
// This file contains the customization implementations of R module
|
|
// to change behavior of libxgboost
|
|
|
|
#include <xgboost/logging.h>
|
|
#include "../../src/common/random.h"
|
|
#include "./xgboost_R.h"
|
|
|
|
// redirect the messages to R's console.
|
|
namespace dmlc {
|
|
void CustomLogMessage::Log(const std::string& msg) {
|
|
Rprintf("%s\n", msg.c_str());
|
|
}
|
|
} // namespace dmlc
|
|
|
|
// implements rabit error handling.
|
|
extern "C" {
|
|
void XGBoostAssert_R(int exp, const char *fmt, ...);
|
|
void XGBoostCheck_R(int exp, const char *fmt, ...);
|
|
}
|
|
|
|
namespace rabit {
|
|
namespace utils {
|
|
extern "C" {
|
|
void (*Printf)(const char *fmt, ...) = Rprintf;
|
|
void (*Assert)(int exp, const char *fmt, ...) = XGBoostAssert_R;
|
|
void (*Check)(int exp, const char *fmt, ...) = XGBoostCheck_R;
|
|
void (*Error)(const char *fmt, ...) = error;
|
|
}
|
|
}
|
|
}
|
|
|
|
namespace xgboost {
|
|
ConsoleLogger::~ConsoleLogger() {
|
|
if (cur_verbosity_ == LogVerbosity::kIgnore ||
|
|
cur_verbosity_ <= global_verbosity_) {
|
|
dmlc::CustomLogMessage::Log(log_stream_.str());
|
|
}
|
|
}
|
|
TrackerLogger::~TrackerLogger() {
|
|
dmlc::CustomLogMessage::Log(log_stream_.str());
|
|
}
|
|
} // namespace xgboost
|
|
|
|
namespace xgboost {
|
|
namespace common {
|
|
|
|
// redirect the nath functions.
|
|
bool CheckNAN(double v) {
|
|
return ISNAN(v);
|
|
}
|
|
#if !defined(XGBOOST_USE_CUDA)
|
|
double LogGamma(double v) {
|
|
return lgammafn(v);
|
|
}
|
|
#endif // !defined(XGBOOST_USE_CUDA)
|
|
// customize random engine.
|
|
void CustomGlobalRandomEngine::seed(CustomGlobalRandomEngine::result_type val) {
|
|
// ignore the seed
|
|
}
|
|
|
|
// use R's PRNG to replacd
|
|
CustomGlobalRandomEngine::result_type
|
|
CustomGlobalRandomEngine::operator()() {
|
|
return static_cast<result_type>(
|
|
std::floor(unif_rand() * CustomGlobalRandomEngine::max()));
|
|
}
|
|
} // namespace common
|
|
} // namespace xgboost
|