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

[dask] Remove the workaround for segfault. (#7146)

Browse Source
This commit is contained in:
Jiaming Yuan 2021-07-30 03:57:53 +08:00 committed by GitHub
parent f1a4a1ac95
commit 1369133916
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
python-package/xgboost/dask.py
View File

@ -1219,8 +1219,7 @@ async def _predict_async(
base_margin = None base_margin = None
for i, blob in enumerate(part[1:]): for i, blob in enumerate(part[1:]):
if meta_names[i] == "base_margin": if meta_names[i] == "base_margin":
# segfault without copy. See https://github.com/dmlc/xgboost/issues/7111. base_margin = blob
base_margin = blob.copy()
with config.config_context(**global_config): with config.config_context(**global_config):
m = DMatrix( m = DMatrix(
data, data,