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

Enforce row-major order in cuPy array (#6459)

Browse Source
This commit is contained in:
Philip Hyunsu Cho 2020-12-03 18:29:10 -08:00 committed by Hyunsu Cho
parent 3efc4ea0d1
commit 3a83fcb0eb
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
python-package/xgboost/data.py
View File

@ -424,6 +424,7 @@ def _transform_cupy_array(data):
data, '__array__'): data, '__array__'):
import cupy # pylint: disable=import-error import cupy # pylint: disable=import-error
data = cupy.array(data, copy=False) data = cupy.array(data, copy=False)
data = data.astype(dtype=data.dtype, order='C', copy=False)
return data return data