From 13691339166820558d5b6aafef6d81d33b0dcce9 Mon Sep 17 00:00:00 2001 From: Jiaming Yuan Date: Fri, 30 Jul 2021 03:57:53 +0800 Subject: [PATCH] [dask] Remove the workaround for segfault. (#7146) --- python-package/xgboost/dask.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/python-package/xgboost/dask.py b/python-package/xgboost/dask.py index 580897ece..9883a0c5b 100644 --- a/python-package/xgboost/dask.py +++ b/python-package/xgboost/dask.py @@ -1219,8 +1219,7 @@ async def _predict_async( base_margin = None for i, blob in enumerate(part[1:]): if meta_names[i] == "base_margin": - # segfault without copy. See https://github.com/dmlc/xgboost/issues/7111. - base_margin = blob.copy() + base_margin = blob with config.config_context(**global_config): m = DMatrix( data,