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 4444//
4545// In this specific test we use M == 4, N == 4, and K == 4.
4646
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+
4753// CHECK-IR-LABEL: llvm.func @matrix_by_matrix_mul_and_acc
4854// CHECK-IR-COUNT-4: arm_neon.intr.bfmmla
4955func.func @matrix_by_matrix_mul_and_acc () {
@@ -110,10 +116,6 @@ func.func @matrix_by_matrix_mul_and_acc() {
110116// * ACC, OUT: vector<Nxf32>
111117// N is even and K is divisible by 4.
112118// 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.
117119
118120// CHECK-IR-LABEL: llvm.func @vector_by_matrix_mul_and_acc
119121// CHECK-IR-COUNT-2: arm_neon.intr.bfmmla
You can’t perform that action at this time.
0 commit comments