Explicitly set PartialReduceMLAActions

Author: NickGuy-Arm

llvm/lib/Target/AArch64/AArch64ISelLowering.cpp

@@ -1603,6 +1603,14 @@ AArch64TargetLowering::AArch64TargetLowering(const TargetMachine &TM,
       // 3. Mark known legal pairs as 'Legal' (these will expand to USDOT).
       setPartialReduceMLAAction(MVT::nxv2i64, MVT::nxv8i16, Legal);
       setPartialReduceMLAAction(MVT::nxv4i32, MVT::nxv16i8, Legal);
+
+      setPartialReduceMLAAction(MVT::nxv2i64, MVT::nxv8i64, Custom);
+      setPartialReduceMLAAction(MVT::nxv4i32, MVT::nxv16i32, Custom);
+
+      setPartialReduceMLAAction(MVT::nxv2i64, MVT::nxv4i64, Custom);
+      setPartialReduceMLAAction(MVT::nxv4i32, MVT::nxv8i32, Custom);
+      setPartialReduceMLAAction(MVT::nxv8i16, MVT::nxv16i16, Custom);
+      setPartialReduceMLAAction(MVT::nxv16i8, MVT::nxv32i8, Custom);
     }
 
     // Firstly, exclude all scalable vector extending loads/truncating stores,