tree 1518d366ca2dea52937c640ea20ac2e74a562f89
parent d5839584547ec2a258b12399f56ea3d006668dbb
author Han-Chung Wang <hanhan0912@gmail.com> 1727367086 -0700
committer GitHub <noreply@github.com> 1727367086 -0700
gpgsig -----BEGIN PGP SIGNATURE-----
 
 wsFcBAABCAAQBQJm9YeuCRC1aQ7uu5UhlAAAgqYQAFS5cHUGSaerDjoNlzjE7Igt
 n9Jo/NHc/KfAaL2cCse2Jox9BeLq23Zbajjxh8ht369wYibjgMoOB9TEz3g0Cyna
 CWVW0QHsaXGm7szwrUWQcD8FU1bOIBPNM2+7fjVFB82Zjt97JbNnvDNiLt7lWdit
 ZJNzry914Ktj2FnOZpNOwk3MHgSttUqSQn6+/8cHKXWNse2Bxwo2OqEvOYMTuz2C
 nEDbskjCMaesregKN7PDoyx6lRzuRVNwim0y3c8XSwVdJC/CinNflSOM0oE16Y+O
 X2RiE63DJFVuzBBjymnyXmV2hDJmNi6pT0QA5bqEiswG3ntt42MU84myZ6OOVzpj
 vN2sP9BvLkQ6JvYs79Mi5G6HxCrhO+TuKtxIjrde+8BoKtXc7+kvIY6dzxLLvA6W
 ACdTbOlaV2DhTUWqhyEQfeD7RDCCHBojnVE9VNaqKNq4pRKcszdmgQe9md7ZWEgl
 wqkJnC0mTwN6jFz9Q7enG2ZIT203c/zS7kfm7wvl4u3geurpGnXNiCtBYTW1OXFd
 oLNQhdgfe99/xPUUTu9Jju0/8I0LdWhNnXLy4xFocrKNbhcbgWGyla7/i5Vnm4sk
 G7W7Fwjz4jHM6+BdBOxal+pUQk729vlTSsvoORuJAa3Z953affb3bhWSOB2kJKwW
 vplmrUfLZSzo9rhLbafA
 =AVdY
 -----END PGP SIGNATURE-----
 

[DT] Collapse matmul_narrow_M/N field into round_dims_to attribute. (#18599)

The revision refines how we handle matmul_narrow_M/N in encodings. The
matmul_narrow_M/N attributes are retired in the patch. The information
is carried in `round_dims_to` attribute which is rounded to the next
power of two. E.g., if the value of `matmul_narrow_M` is `3`, it will be
`4` in the `round_dims_to` attribute.

The `kNarrowThreshold` constant is introduced to Encoding namespace. A
dimension is considered as "narrow" only if the size of the dimension is
less than `kNarrowThreshold`.

The revision implements `getMatmulNarrowM` and `getMatmulNarrowN`
helpers based on the definition of `round_dims_to` and the
`kNarrowThreshold` variable. Users can call the helpers to get the
narrow size of M/N. Again, you'll get the value that is rounded to the
next power of two if they are considered "narrow". E.g.,
`getMatmulNarrowM()` returns `4` if `M` equals to `3`.

The revision also fixes a bug in `chooseMatmulTile` on CPU side. The
hostDefinedUpperBound is not updated in the narrow-N cases.

Fixes https://github.com/iree-org/iree/issues/17729
Fixes https://github.com/iree-org/iree/issues/15466

---------

Signed-off-by: hanhanW <hanhan0912@gmail.com>