@@ -49,16 +49,16 @@ def test_gemv():
4949 bools = [False , True ]
5050 for shape , order , trans , offseted_i , sliced in product (
5151 [(100 , 128 ), (128 , 50 )], 'fc' , bools , bools , [1 , 2 , - 1 , - 2 ]):
52- yield gemv , shape , 'float32' , order , trans , \
53- offseted_i , sliced , True , False
52+ yield ( gemv , shape , 'float32' , order , trans ,
53+ offseted_i , sliced , True , False )
5454 for overwrite , init_y in product (bools , bools ):
55- yield gemv , (4 , 3 ), 'float32' , 'f' , False , False , 1 , \
56- overwrite , init_y
55+ yield ( gemv , (4 , 3 ), 'float32' , 'f' , False , False , 1 ,
56+ overwrite , init_y )
5757 yield gemv , (32 , 32 ), 'float64' , 'f' , False , False , 1 , True , False
5858 for alpha , beta , overwrite in product (
5959 [0 , 1 , - 1 , 0.6 ], [0 , 1 , - 1 , 0.6 ], bools ):
60- yield gemv , (32 , 32 ), 'float32' , 'f' , False , False , 1 , \
61- overwrite , True , alpha , beta
60+ yield ( gemv , (32 , 32 ), 'float32' , 'f' , False , False , 1 ,
61+ overwrite , True , alpha , beta )
6262
6363
6464@guard_devsup
@@ -96,17 +96,17 @@ def test_gemm():
9696 for (m , n , k ), order , trans , offseted_o in product (
9797 [(48 , 15 , 32 ), (15 , 32 , 48 )], list (product (* ['fc' ]* 3 )),
9898 list (product (bools , bools )), bools ):
99- yield gemm , m , n , k , 'float32' , order , trans , \
100- offseted_o , 1 , False , False
99+ yield ( gemm , m , n , k , 'float32' , order , trans ,
100+ offseted_o , 1 , False , False )
101101 for sliced , overwrite , init_res in product ([1 , 2 , - 1 , - 2 ], bools , bools ):
102- yield gemm , 4 , 3 , 2 , 'float32' , ('f' , 'f' , 'f' ), \
103- (False , False ), False , sliced , overwrite , init_res
104- yield gemm , 32 , 32 , 32 , 'float64' , ('f' , 'f' , 'f' ), (False , False ), \
105- False , 1 , False , False
102+ yield ( gemm , 4 , 3 , 2 , 'float32' , ('f' , 'f' , 'f' ),
103+ (False , False ), False , sliced , overwrite , init_res )
104+ yield ( gemm , 32 , 32 , 32 , 'float64' , ('f' , 'f' , 'f' ), (False , False ),
105+ False , 1 , False , False )
106106 for alpha , beta , overwrite in product (
107107 [0 , 1 , - 1 , 0.6 ], [0 , 1 , - 1 , 0.6 ], bools ):
108- yield gemm , 32 , 23 , 32 , 'float32' , ('f' , 'f' , 'f' ), \
109- (False , False ), False , 1 , overwrite , True , alpha , beta
108+ yield ( gemm , 32 , 23 , 32 , 'float32' , ('f' , 'f' , 'f' ),
109+ (False , False ), False , 1 , overwrite , True , alpha , beta )
110110
111111
112112@guard_devsup
@@ -179,17 +179,17 @@ def test_rgemmBatch_3d():
179179 [1 , 17 , 31 ], [(24 , 7 , 16 ), (7 , 16 , 24 )],
180180 list (product ('fc' , 'fc' , 'c' )),
181181 list (product (bools , bools )), bools ):
182- yield rgemmBatch_3d , b , m , n , k , 'float32' , order , trans , \
183- offseted_o , 1 , False , False
182+ yield ( rgemmBatch_3d , b , m , n , k , 'float32' , order , trans ,
183+ offseted_o , 1 , False , False )
184184 for sliced , overwrite , init_res in product ([1 , 2 , - 1 , - 2 ], bools , bools ):
185- yield rgemmBatch_3d , 5 , 4 , 3 , 2 , 'float32' , ('f' , 'f' , 'c' ), \
186- (False , False ), False , sliced , overwrite , init_res
187- yield rgemmBatch_3d , 16 , 16 , 16 , 16 , 'float64' , ('f' , 'f' , 'c' ), \
188- (False , False ), False , 1 , False , False
185+ yield ( rgemmBatch_3d , 5 , 4 , 3 , 2 , 'float32' , ('f' , 'f' , 'c' ),
186+ (False , False ), False , sliced , overwrite , init_res )
187+ yield ( rgemmBatch_3d , 16 , 16 , 16 , 16 , 'float64' , ('f' , 'f' , 'c' ),
188+ (False , False ), False , 1 , False , False )
189189 for alpha , beta , overwrite in product (
190190 [0 , 1 , - 1 , 0.6 ], [0 , 1 , - 1 , 0.6 ], bools ):
191- yield rgemmBatch_3d , 16 , 16 , 9 , 16 , 'float32' , ('f' , 'f' , 'c' ), \
192- (False , False ), False , 1 , overwrite , True , alpha , beta
191+ yield ( rgemmBatch_3d , 16 , 16 , 9 , 16 , 'float32' , ('f' , 'f' , 'c' ),
192+ (False , False ), False , 1 , overwrite , True , alpha , beta )
193193
194194
195195@guard_devsup
0 commit comments