@@ -58,7 +58,7 @@ func.func @matrix_by_matrix_mul_and_acc() {
58
58
[ 0.5 , -1.3 , -2.2 , 0.1 ],
59
59
[-0.7 , 1.0 , 1.7 , -1.0 ]]> : vector <4 x4 xf32 >
60
60
61
- %acc_mem = memref.alloc () : memref <4 x4 xf32 >
61
+ %acc_mem = memref.alloca () : memref <4 x4 xf32 >
62
62
vector.transfer_write %acc_cst , %acc_mem [%c0 , %c0 ] {in_bounds = [true , true ] } : vector <4 x4 xf32 >, memref <4 x4 xf32 >
63
63
%acc = vector.transfer_read %acc_mem [%c0 , %c0 ], %c0_f32 {in_bounds = [true , true ]} : memref <4 x4 xf32 >, vector <4 x4 xf32 >
64
64
@@ -68,7 +68,7 @@ func.func @matrix_by_matrix_mul_and_acc() {
68
68
[-0.4 , 0.6 , 0.8 , -0.5 ],
69
69
[-0.6 , -1.0 , -1.0 , -1.0 ]]> : vector <4 x4 xbf16 >
70
70
71
- %lhs_mem = memref.alloc () : memref <4 x4 xbf16 >
71
+ %lhs_mem = memref.alloca () : memref <4 x4 xbf16 >
72
72
vector.transfer_write %lhs_cst , %lhs_mem [%c0 , %c0 ] {in_bounds = [true , true ] } : vector <4 x4 xbf16 >, memref <4 x4 xbf16 >
73
73
%lhs = vector.transfer_read %lhs_mem [%c0 , %c0 ], %c0_bf16 {in_bounds = [true , true ]} : memref <4 x4 xbf16 >, vector <4 x4 xbf16 >
74
74
@@ -78,7 +78,7 @@ func.func @matrix_by_matrix_mul_and_acc() {
78
78
[-0.2 , 0.4 , 1.0 , 0.4 ],
79
79
[-1.3 , -0.2 , -2.2 , 0.3 ]]> : vector <4 x4 xbf16 >
80
80
81
- %rhs_mem = memref.alloc () : memref <4 x4 xbf16 >
81
+ %rhs_mem = memref.alloca () : memref <4 x4 xbf16 >
82
82
vector.transfer_write %rhs_cst , %rhs_mem [%c0 , %c0 ] {in_bounds = [true , true ] } : vector <4 x4 xbf16 >, memref <4 x4 xbf16 >
83
83
%rhs = vector.transfer_read %rhs_mem [%c0 , %c0 ], %c0_bf16 {in_bounds = [true , true ]} : memref <4 x4 xbf16 >, vector <4 x4 xbf16 >
84
84
@@ -121,14 +121,14 @@ func.func @vector_by_matrix_mul_and_acc() {
121
121
// Accumulator test data
122
122
%acc_cst = arith.constant dense <[0.7 , 1.0 , -0.1 , 1.8 ]> : vector <4 xf32 >
123
123
124
- %acc_mem = memref.alloc () : memref <4 xf32 >
124
+ %acc_mem = memref.alloca () : memref <4 xf32 >
125
125
vector.transfer_write %acc_cst , %acc_mem [%c0 ] {in_bounds = [true ] } : vector <4 xf32 >, memref <4 xf32 >
126
126
%acc = vector.transfer_read %acc_mem [%c0 ], %c0_f32 {in_bounds = [true ]} : memref <4 xf32 >, vector <4 xf32 >
127
127
128
128
// LHS test data
129
129
%lhs_cst = arith.constant dense <[0.1 , 0.7 , -0.9 , 1.3 ]> : vector <4 xbf16 >
130
130
131
- %lhs_mem = memref.alloc () : memref <4 xbf16 >
131
+ %lhs_mem = memref.alloca () : memref <4 xbf16 >
132
132
vector.transfer_write %lhs_cst , %lhs_mem [%c0 ] {in_bounds = [true ] } : vector <4 xbf16 >, memref <4 xbf16 >
133
133
%lhs = vector.transfer_read %lhs_mem [%c0 ], %c0_bf16 {in_bounds = [true ]} : memref <4 xbf16 >, vector <4 xbf16 >
134
134
@@ -138,7 +138,7 @@ func.func @vector_by_matrix_mul_and_acc() {
138
138
[-0.2 , 0.4 , 1.0 , 0.4 ],
139
139
[-1.3 , -0.2 , -2.2 , 0.3 ]]> : vector <4 x4 xbf16 >
140
140
141
- %rhs_mem = memref.alloc () : memref <4 x4 xbf16 >
141
+ %rhs_mem = memref.alloca () : memref <4 x4 xbf16 >
142
142
vector.transfer_write %rhs_cst , %rhs_mem [%c0 , %c0 ] {in_bounds = [true , true ] } : vector <4 x4 xbf16 >, memref <4 x4 xbf16 >
143
143
%rhs = vector.transfer_read %rhs_mem [%c0 , %c0 ], %c0_bf16 {in_bounds = [true , true ]} : memref <4 x4 xbf16 >, vector <4 x4 xbf16 >
144
144
0 commit comments