From c103ec51d8508d45fe86125db9a677529af33b9b 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