Remove RABIT CMake targets. (#6275)

* Now it's built as part of libxgboost.
* Set correct C API error in RABIT initialization and finalization.
* Remove redundant message.
* Guard the tracker print C API.
This commit is contained in:
Jiaming Yuan
2020-10-27 01:30:20 +08:00
committed by GitHub
parent 2686d32a36
commit d61b628bf5
11 changed files with 50 additions and 120 deletions

View File

@@ -6,7 +6,7 @@ import ctypes
import pickle
import numpy as np
from .core import _LIB, c_str, STRING_TYPES
from .core import _LIB, c_str, STRING_TYPES, _check_call
def _init_rabit():
@@ -77,7 +77,7 @@ def tracker_print(msg):
msg = str(msg)
is_dist = _LIB.RabitIsDistributed()
if is_dist != 0:
_LIB.RabitTrackerPrint(c_str(msg))
_check_call(_LIB.RabitTrackerPrint(c_str(msg)))
else:
sys.stdout.write(msg)
sys.stdout.flush()