[llvm] [DAG] Reducing instructions by better legalization handling of AVGFLOORU for i128 data types (PR #99913)

Julius Alexandre via llvm-commits llvm-commits at lists.llvm.org
Mon Jul 22 14:37:29 PDT 2024


================
@@ -9352,6 +9353,27 @@ SDValue TargetLowering::expandAVG(SDNode *N, SelectionDAG &DAG) const {
     }
   }
 
+  if (Opc == ISD::AVGFLOORU && SVT == MVT::i128) {
+    SDValue UAddWithOverflow = DAG.getNode(ISD::UADDO, dl, 
+                                           DAG.getVTList(VT, MVT::i1), { RHS, LHS });
+
+    SDValue Sum = UAddWithOverflow.getValue(0);
+    SDValue Overflow = UAddWithOverflow.getValue(1);
+
+    // Right shift the sum by 1
+    SDValue One = DAG.getConstant(1, dl, VT);
+    SDValue LShrVal = DAG.getNode(ISD::SRL, dl, VT, Sum, One);
----------------
medievalghoul wrote:

Just to clarify, i would just literally do 
`One = DAG.getShiftAmountConstant(1, VT, dl);` instead? or did you mean something else

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


More information about the llvm-commits mailing list