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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
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");
} }
@ -45,7 +48,7 @@ void AllreduceBase::Init(void) {
if (attempt_id != 0) { if (attempt_id != 0) {
const char *att = strrchr(attempt_id, '_'); const char *att = strrchr(attempt_id, '_');
int num_trial; int num_trial;
if (att != NULL && sscanf(att+1, "%d", &num_trial) == 1) { if (att != NULL && sscanf(att + 1, "%d", &num_trial) == 1) {
this->SetParam("rabit_num_trial", att + 1); this->SetParam("rabit_num_trial", att + 1);
} }
} }