Support 'a' and 'k'

cupy · Jun 19, 2018 · 50e9150 · 50e9150
1 parent 1faceab
commit 50e9150
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cupy/core/core.pyx b/cupy/core/core.pyx
@@ -2121,7 +2121,7 @@ cpdef ndarray array(obj, dtype=None, bint copy=True, str order='K',
                 a = a.view()
             a.shape = (1,) * (ndmin - ndim) + a.shape
     else:
-        if order == 'K' or order == 'A':
+        if order is not None and len(order) == 1 and order in 'KAka':
             if isinstance(obj, numpy.ndarray) and obj.flags.f_contiguous:
                 order = 'F'
             else: