[llvm] [SelectionDAG][AArch64] Add dot product lowering in NEON for PARTIAL_REDUCE_*MLA ISD nodes (PR #140075)
Benjamin Maxwell via llvm-commits
llvm-commits at lists.llvm.org
Thu May 15 09:00:27 PDT 2025
================
@@ -7743,8 +7752,11 @@ SDValue AArch64TargetLowering::LowerOperation(SDValue Op,
case ISD::EXPERIMENTAL_VECTOR_HISTOGRAM:
return LowerVECTOR_HISTOGRAM(Op, DAG);
case ISD::PARTIAL_REDUCE_SMLA:
- case ISD::PARTIAL_REDUCE_UMLA:
- return LowerPARTIAL_REDUCE_MLA(Op, DAG);
+ case ISD::PARTIAL_REDUCE_UMLA: {
+ if (SDValue Result = LowerPARTIAL_REDUCE_MLA(Op, DAG))
+ return Result;
+ return expandPartialReduceMLA(Op.getNode(), DAG);
----------------
MacDue wrote:
Ditto here:
```suggestion
return LowerPARTIAL_REDUCE_MLA(Op, DAG);
```
https://github.com/llvm/llvm-project/pull/140075
More information about the llvm-commits
mailing list