@@ -517,6 +517,11 @@ def copy(usm_ary, order="K"):
517
517
- "K": match the layout of `usm_ary` as closely as possible.
518
518
519
519
"""
520
+ if len (order ) == 0 or order [0 ] not in "KkAaCcFf" :
521
+ raise ValueError (
522
+ "Unrecognized order keyword value, expecting 'K', 'A', 'F', or 'C'."
523
+ )
524
+ order = order [0 ].upper ()
520
525
if not isinstance (usm_ary , dpt .usm_ndarray ):
521
526
return TypeError (
522
527
f"Expected object of type dpt.usm_ndarray, got { type (usm_ary )} "
@@ -585,11 +590,11 @@ def astype(usm_ary, newdtype, order="K", casting="unsafe", copy=True):
585
590
return TypeError (
586
591
f"Expected object of type dpt.usm_ndarray, got { type (usm_ary )} "
587
592
)
588
- if not isinstance (order , str ) or order not in [ "A" , "C" , "F" , "K" ] :
593
+ if len (order ) == 0 or order [ 0 ] not in "KkAaCcFf" :
589
594
raise ValueError (
590
- "Unrecognized value of the order keyword. "
591
- "Recognized values are 'A', 'C', 'F', or 'K'"
595
+ "Unrecognized order keyword value, expecting 'K', 'A', 'F', or 'C'."
592
596
)
597
+ order = order [0 ].upper ()
593
598
ary_dtype = usm_ary .dtype
594
599
target_dtype = _get_dtype (newdtype , usm_ary .sycl_queue )
595
600
if not dpt .can_cast (ary_dtype , target_dtype , casting = casting ):
0 commit comments