@@ -1198,12 +1198,15 @@ def test_syrk(self, dt):
1198
1198
assert result is iout
1199
1199
assert_dtype_allclose (result , expected )
1200
1200
1201
+ result = ia .mT @ ia
1202
+ expected = a .T @ a
1203
+ assert_dtype_allclose (result , expected )
1204
+
1201
1205
@pytest .mark .parametrize (
1202
1206
"order, out_order" ,
1203
1207
[("C" , "C" ), ("C" , "F" ), ("F" , "C" ), ("F" , "F" )],
1204
1208
)
1205
1209
def test_syrk_out_order (self , order , out_order ):
1206
- # test syrk with out keyword
1207
1210
a = generate_random_numpy_array ((5 , 4 ), order = order , low = - 5 , high = 5 )
1208
1211
out = numpy .empty ((5 , 5 ), dtype = a .dtype , order = out_order )
1209
1212
ia , iout = dpnp .array (a ), dpnp .array (out )
@@ -1215,6 +1218,14 @@ def test_syrk_out_order(self, order, out_order):
1215
1218
assert result .flags .f_contiguous == expected .flags .f_contiguous
1216
1219
assert_dtype_allclose (result , expected )
1217
1220
1221
+ @pytest .mark .parametrize ("order" , ["F" , "C" ])
1222
+ def test_syrk_order (self , order ):
1223
+ a = generate_random_numpy_array ((4 , 6 ), order = order , low = - 5 , high = 5 )
1224
+ ia = dpnp .array (a )
1225
+ expected = numpy .matmul (a , a .T )
1226
+ result = dpnp .matmul (ia , ia .mT )
1227
+ assert_dtype_allclose (result , expected )
1228
+
1218
1229
def test_bool (self ):
1219
1230
a = generate_random_numpy_array ((3 , 4 ), dtype = dpnp .bool )
1220
1231
b = generate_random_numpy_array ((4 , 5 ), dtype = dpnp .bool )
0 commit comments