File tree Expand file tree Collapse file tree 1 file changed +7
-2
lines changed
mlir/test/Integration/Dialect/Vector/CPU/ArmSVE Expand file tree Collapse file tree 1 file changed +7
-2
lines changed Original file line number Diff line number Diff line change @@ -68,8 +68,13 @@ func.func @dot_product_i32() {
68
68
// on the vector length, so we are unable to verify it.
69
69
%dp1 = vector.contract #dotp_trait %vector_a , %vector_b , %acc
70
70
: vector <[4 ]xi32 >, vector <[4 ]xi32 > into i32
71
- // DP: {{[0-9]*}}
72
- vector.print %dp1 : i32
71
+ // Dot product should be (123 * 314) * 4 * vscale, so ...
72
+ %vscale = vector.vscale
73
+ %vscale_i32 = arith.index_cast %vscale : index to i32
74
+ %dp1_divvl = arith.divui %dp1 , %vscale_i32 : i32
75
+ // ... %dp/%vscale = 123 * 314 * 4 = 154488
76
+ // DP: 154488
77
+ vector.print %dp1_divvl : i32
73
78
74
79
// The result of this dot-product should be 0.
75
80
%dp2 = vector.contract #dotp_trait %vector_a , %vector_c , %acc
You can’t perform that action at this time.
0 commit comments