From 3a83fcb0eb987d2aa37737640c379e21edd07a76 Mon Sep 17 00:00:00 2001 From: Philip Hyunsu Cho Date: Thu, 3 Dec 2020 18:29:10 -0800 Subject: [PATCH] Enforce row-major order in cuPy array (#6459) --- python-package/xgboost/data.py | 1 + 1 file changed, 1 insertion(+) diff --git a/python-package/xgboost/data.py b/python-package/xgboost/data.py index 2705249b7..2babeafea 100644 --- a/python-package/xgboost/data.py +++ b/python-package/xgboost/data.py @@ -424,6 +424,7 @@ def _transform_cupy_array(data): data, '__array__'): import cupy # pylint: disable=import-error data = cupy.array(data, copy=False) + data = data.astype(dtype=data.dtype, order='C', copy=False) return data