[llvm] [AArch64][SVE] Add dot product lowering for PARTIAL_REDUCE_MLA node (PR #130933)

[Benjamin Maxwell via llvm-commits]

================
@@ -1585,6 +1585,26 @@ AArch64TargetLowering::AArch64TargetLowering(const TargetMachine &TM,
       setOperationAction(ISD::MSTORE, VT, Custom);
     }
 
+    if (EnablePartialReduceNodes) {
+      for (MVT VT : MVT::integer_scalable_vector_valuetypes()) {
+        for (MVT InnerVT : MVT::integer_scalable_vector_valuetypes()) {
+          // 1. Set all combinations where a type is illegal to "Legal"
+          // - These will be legalized to a legal type pair
+          // - Avoid expanding them too early (or preventing folds)
+          if (!isTypeLegal(VT) || !isTypeLegal(InnerVT)) {
+            setPartialReduceMLAAction(VT, InnerVT, Legal);
+            continue;
+          }
+          //  2. Set all legal combinations to "Expand"
+          // - Not all of these can be lowered (via a Legal or Custom lowering).
+          setPartialReduceMLAAction(VT, InnerVT, Expand);
+        }
+      }
+      // 3. Mark known legal pairs as 'Legal' (these will expand to USDOT).
----------------
MacDue wrote:

The comment here is referring to SDOT or UDOT not USDOT (my bad). I don't think we should include extension info, as the node does not include that information (so it would not be available where `getPartialReduceMLAAction` is used without looking at operands UUIC). I think the USDOT case can be handled by marking applicable type pairs as "Custom", the custom lowering can then handle producing USDOTs when applicable. I believe if a custom lowering just returns "SDValue()" the legalizer assumes the node is legal, so we can still use the standard ISEL for non-USDOTs. 

https://github.com/llvm/llvm-project/pull/130933