[llvm] [AArch64][SVE] Add dot product lowering for PARTIAL_REDUCE_MLA node (PR #130933)
Nicholas Guy via llvm-commits
llvm-commits at lists.llvm.org
Wed Mar 26 07:38:15 PDT 2025
================
@@ -2712,6 +2736,20 @@ class TargetLoweringBase {
setCondCodeAction(CCs, VT, Action);
}
+ /// Indicate how a PARTIAL_REDUCE_U/SMLA node with Acc type AccVT and Input
+ /// type InputVT should be treated by the target. Either it's legal, needs to
+ /// be promoted to a larger size, needs to be expanded to some other code
+ /// sequence, or the target has a custom expander for it.
+ void setPartialReduceMLAAction(MVT AccVT, MVT InputVT,
+ LegalizeAction Action) {
+ assert(AccVT.isValid() && InputVT.isValid() &&
----------------
NickGuy-Arm wrote:
I'd rather not silently ignore calls to this, even if they could be a nop in some cases.
In the (admittedly unlikely) case something tries to override an assigned action (think assigned to something in a loop, then reassigned to `Expand` following said loop), the second call would be ignored.
https://github.com/llvm/llvm-project/pull/130933
More information about the llvm-commits
mailing list