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

add support for yarn

Browse Source
This commit is contained in:
tqchen 2014-12-22 03:24:00 -08:00
parent d3433c5946
commit ab7492dbc2
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/allreduce_base.cc
View File

@ -43,6 +43,9 @@ void AllreduceBase::Init(void) {
}
// handling for hadoop
const char *num_task = getenv("mapred_map_tasks");
if (num_task == NULL) {
num_task = getenv("mapreduce_job_maps");
}
if (hadoop_mode != 0) {
utils::Check(num_task != NULL, "hadoop_mode is set but cannot find mapred_map_tasks");
}