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

add task id

Browse Source
This commit is contained in:
tqchen
2014-12-22 04:17:23 -08:00
parent 0dd51d5dd0
commit 15836eb98e
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/allreduce_base.cc
View File

@@ -34,6 +34,9 @@ void AllreduceBase::Init(void) {
// setup from enviroment variables // setup from enviroment variables
{// handling for hadoop {// handling for hadoop
const char *task_id = getenv("mapred_tip_id"); const char *task_id = getenv("mapred_tip_id");
if (task_id == NULL) {
task_id = getenv("mapreduce_task_id");
}
if (hadoop_mode != 0) { if (hadoop_mode != 0) {
utils::Check(task_id != NULL, "hadoop_mode is set but cannot find mapred_task_id"); utils::Check(task_id != NULL, "hadoop_mode is set but cannot find mapred_task_id");
} }