@@ -922,10 +922,10 @@ def test_can_cast():
922
922
q = get_queue_or_skip ()
923
923
924
924
# incorrect input
925
- X = dpt .ones ((2 , 2 ), dtype = dpt .int64 , sycl_queue = q )
925
+ X = dpt .ones ((2 , 2 ), dtype = dpt .int16 , sycl_queue = q )
926
926
pytest .raises (TypeError , dpt .can_cast , X , 1 )
927
927
pytest .raises (TypeError , dpt .can_cast , X , X )
928
- X_np = np .ones ((2 , 2 ), dtype = np .int64 )
928
+ X_np = np .ones ((2 , 2 ), dtype = np .int16 )
929
929
930
930
assert dpt .can_cast (X , "float32" ) == np .can_cast (X_np , "float32" )
931
931
assert dpt .can_cast (X , dpt .int32 ) == np .can_cast (X_np , np .int32 )
@@ -935,8 +935,8 @@ def test_can_cast():
935
935
def test_result_type ():
936
936
q = get_queue_or_skip ()
937
937
938
- X = [dpt .ones ((2 ), dtype = dpt .int64 , sycl_queue = q ), dpt .int32 , "float16 " ]
939
- X_np = [np .ones ((2 ), dtype = np .int64 ), np .int32 , "float16 " ]
938
+ X = [dpt .ones ((2 ), dtype = dpt .int16 , sycl_queue = q ), dpt .int32 , "int64 " ]
939
+ X_np = [np .ones ((2 ), dtype = np .int16 ), np .int32 , "int64 " ]
940
940
941
941
assert dpt .result_type (* X ) == np .result_type (* X_np )
942
942
0 commit comments