[llvm] [WIP][SDAG] Add partial_reduce_sumla node (PR #141267)
Paul Walker via llvm-commits
llvm-commits at lists.llvm.org
Tue May 27 04:43:26 PDT 2025
================
@@ -1874,8 +1874,8 @@ AArch64TargetLowering::AArch64TargetLowering(const TargetMachine &TM,
if (EnablePartialReduceNodes && Subtarget->isSVEorStreamingSVEAvailable()) {
// Mark known legal pairs as 'Legal' (these will expand to UDOT or SDOT).
// Other pairs will default to 'Expand'.
- setPartialReduceMLAAction(MVT::nxv2i64, MVT::nxv8i16, Legal);
- setPartialReduceMLAAction(MVT::nxv4i32, MVT::nxv16i8, Legal);
+ setPartialReduceMLAAction(MVT::nxv2i64, MVT::nxv8i16, Custom);
----------------
paulwalker-arm wrote:
Just throwing this out there but perhaps change it to `setReduceAction(Opcode, ResultType, OperandType)`? to make it easier to add more reductions in the future. This might also be a way to relax the result type requirements of the current `VECREDUCE_***` nodes.
https://github.com/llvm/llvm-project/pull/141267
More information about the llvm-commits
mailing list