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

2020-12-03 18:29:10 -08:00 · 2020-12-03 18:29:10 -08:00 · 3a83fcb0eb
commit 3a83fcb0eb
parent 3efc4ea0d1
1 changed files with 1 additions and 0 deletions
--- 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