File tree Expand file tree Collapse file tree 1 file changed +6
-4
lines changed
mlir/test/Integration/Dialect/Vector/CPU/ArmNeon Expand file tree Collapse file tree 1 file changed +6
-4
lines changed Original file line number Diff line number Diff line change 44
44
//
45
45
// In this specific test we use M == 4, N == 4, and K == 4.
46
46
47
+ // Note: In this and in the following test the seemingly unnecessary
48
+ // writes of test vectors to memory are done in order to introduce memory
49
+ // load operations on the path leading up to `vector.contract` since
50
+ // that's more representation of the typical usage when multiplying
51
+ // big, non-constant tensors.
52
+
47
53
// CHECK-IR-LABEL: llvm.func @matrix_by_matrix_mul_and_acc
48
54
// CHECK-IR-COUNT-4: arm_neon.intr.bfmmla
49
55
func.func @matrix_by_matrix_mul_and_acc () {
@@ -110,10 +116,6 @@ func.func @matrix_by_matrix_mul_and_acc() {
110
116
// * ACC, OUT: vector<Nxf32>
111
117
// N is even and K is divisible by 4.
112
118
// In this specific test we use N == 4, and K == 4.
113
- // Note: the seemingly unnecessary writes of test vectors to memory are done
114
- // in order to introduce memory load operations on the path leading up to
115
- // `vector.contract` since that's more representation of the typical usage
116
- // when multiplying big, non-constant tensors.
117
119
118
120
// CHECK-IR-LABEL: llvm.func @vector_by_matrix_mul_and_acc
119
121
// CHECK-IR-COUNT-2: arm_neon.intr.bfmmla
You can’t perform that action at this time.
0 commit comments