tree ad872da677795ab18b8013e0f3c09024b02f47b2
parent 1a626e63b5a009075eea87c01f0661144b1ec010
author Artemiy Bulavin <artemiyb@graphcore.ai> 1741799809 -0400
committer GitHub <noreply@github.com> 1741799809 +0000
gpgsig -----BEGIN PGP SIGNATURE-----
 
 wsFcBAABCAAQBQJn0cGBCRC1aQ7uu5UhlAAAfZkQAHi9sl+KkevMP6ZN5lmNH1mI
 qD9WnkTGInWI8pYxnEFkoYQKSXFV0wcyDz0W+Q+GbddbtySTvNZ+YJA2pmVvW1O/
 kGkmzxeLSamnimvlX1RXOoEpy0qgq1Kn+Av/rk4by1PZuKEVH66wxVtPr3/72b1d
 jdzF8jfOq0GxBMFUYJM6MlLY8hdN4vtXJrY5cDPyXQWX03mWgcci19qEuqlHBMNo
 3PU09uW/EGnk/Ipo99fsizts/NsW1K3T+YMefzdvuItvkPtYKUJZa0l/LspIHIWT
 5fStyZw7ZbA6VudFqZeBQqu5wGXNq+GOAn26xina5woqniFfksJ8Tqqw1jnAjS+/
 EHl8a8yjRXci9PM7iWGbRBfKRS2G1FV0bpplT8oCgK2z9EyaQqSKMZm8dltPgnlI
 wiq8NtX/8RhKQuo9sLQ92guHjdkww23jIG8CM+OiZpGbgCgj+4whAkH2kT/QI8nP
 aNEaRQ6BF/lZEZfkLPgATBsnELqDeoDYLj/sLw8J4OEhkuiHMoab/8EAoXxIPlc2
 RAmoHmYsZUC3ZlR+jEGI0DJz014ZYvkRG9sZkoQ5ZlmJ+jVZ/3qnfa4qQ8xdxjgN
 b6oGdteAUyIXCWGF2vHpgyb3HY6/EawNM4zj6MmUM7rDav24SkWNnpiAp5rifhUI
 HFKk3rpddh7EZVWrCvZJ
 =DVN7
 -----END PGP SIGNATURE-----
 

[mlir] Extract RHS rows once when lowering vector.contract to dot (#130130)

The `vector.contract` op on two matrices A and B will be lowered to
individual dot products of each row and column of A and B respectively.
The existing lowering will extract each column of B for each row of A,
which leads to multiple values in the IR representing the same columns
of B.

This PR makes changes to the `ContractOpToDotLowering` to make sure that
the columns of B are only ever extracted once, so then the SSA values
representing the extracted columns are then re-used in the IR for later
dot products.

I have updated the existing vector-contract-to-dot-transforms test.