Skip to content

Commit e9a91c0

Browse files
[fixup] Add a comment about memory ops
1 parent 8dfd754 commit e9a91c0

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

mlir/test/Integration/Dialect/Vector/CPU/ArmNeon/vector-contract-bfmmla.mlir

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -110,6 +110,10 @@ func.func @matrix_by_matrix_mul_and_acc() {
110110
// * ACC, OUT: vector<Nxf32>
111111
// N is even and K is divisible by 4.
112112
// 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.
113117

114118
// CHECK-IR-LABEL: llvm.func @vector_by_matrix_mul_and_acc
115119
// CHECK-IR-COUNT-2: arm_neon.intr.bfmmla

0 commit comments

Comments
 (0)