tree 5ee2df6b15117580c723127b2363d0f34b03a207
parent 7b9aa284500d1cef0711965ca5b45efd43ce9742
author Zhuoran Yin <zhuoryin@amd.com> 1736361424 -0500
committer GitHub <noreply@github.com> 1736361424 -0500
gpgsig -----BEGIN PGP SIGNATURE-----
 
 wsFcBAABCAAQBQJnfsXQCRC1aQ7uu5UhlAAABNEQAHsv+DA5YFfotW+MZSQz8/Q8
 9HhnYPanPQCyfu3/23iyN8p0Bof3D/hSZzs3SeNpyTivRPzfAZ+buZUQOmgDXx1E
 JeHVyakbcCYLGcevENdrFO27GW5z2dNslRP9opVbXV6BQDv8y05xlteZ8fjtkQD2
 sXrDYrJi4ntaRn6uK4aOmiI2Qr83fvNMkjdFgPxpiXKUApjkW2Q0ZkFl/qZdTYb2
 gCvn6xuVC0ynLJ8QkRezlxiqMdhVG7sxk3wnYCqlr8a8j8gFwvf6rgUXYPL0CNUo
 QHtASQqYoUHN/6q6d/xpL5zCTi01TH8jjjQnQtQEtkVokeMjwPzOKdin5AatI3MD
 KbRJW+tqzADThSK16t0z68akELW7Kdwh3aVfHagz2yetEnPwQpSU/M0SpjifCX+x
 yxeVSDbB9n8U+sGdu3MaDwmkbggOMuGMYeQPzNsBsemqXzKHAwDt60Drux+MorH2
 nRORhXiBKeU/VthUDVJVFyhphjRl7A6J8pq0NTONCbmkddbqIl+e0k2qQPRslj+Y
 6++kFG8WgL+RL5m2CVB0fFX7vamFetZ0D7oPSZJBpxfXjEI5RBjDCZSEtIieELVn
 y/YXB+hAB4PuHlAn8wvWwq5mzu0pHrpAxKt1rOzBJjn1HwBlL0yDw4yaIhVQ6/MW
 bKyMpZ45q8GCX3B+lOeG
 =cCPe
 -----END PGP SIGNATURE-----
 

[GPU][Codegen] Allowing mfma for narrow problem config sizes (#19615)

The motivation of this PR is convolution performance for resnet50
configs. With this PR (and a few pending ones), conv performance with
igemm pipeline get decent speedup in situation where a standalone
dimension size is smaller than intrinsic size. (Take dispatch 69 as
example, the select tile m:7, n:512, k:4608 will be rejected from mfma
because m tile is smaller than intrinsic size of 16). This happens
because previously we are too defensive about when to use intrinsic: in
situation when alignment is not required, we still enforce mfma to be
picked up only when m/n/k tiles are all larger than intrinsic size.

With @nirvedhmeshram's https://github.com/iree-org/iree/pull/19271 and
https://github.com/iree-org/iree/pull/19484, padding is allowed in tile
and fuse matmul and igemm tile and fuse pipelines, it is no longer
necessary to be as conservative as before. I am therefore getting rid of
the conditional check that blocks mfma from being picked up.

This will impact a few pipelines that use `canTargetIntrinsic()`:
- `LLVMGPUPadAndVectorDistribute` will allow narrow m/n/k dimension
sizes for batch matmul
- In `iree-codegen-rocdl-configuration-pipeline`, will allow narrow
m/n/k dimension sizes for matmul (instead of warp reduction)

---------

Signed-off-by: jerryyin <zhuoryin@amd.com>