[llvm] [DAG] Added m_AnyBinOp and m_c_AnyBinOp in SDPatternMatch.h (PR #86435)

Min-Yih Hsu via llvm-commits llvm-commits at lists.llvm.org
Wed Mar 27 08:57:56 PDT 2024


================
@@ -457,6 +457,53 @@ struct BinaryOpc_match {
   }
 };
 
+template <typename LHS_t, typename RHS_t, typename PredFuncT>
+struct AnyBinaryOp_match {
+  LHS_t L;
+  RHS_t R;
+  PredFuncT PredFunc;
+  bool Commutable;
+
+  AnyBinaryOp_match(const PredFuncT &Pred, const LHS_t &LHS, const RHS_t &RHS,
+                    const bool Commutable)
+      : PredFunc(Pred), L(LHS), R(RHS), Commutable(Commutable) {}
+
+  template <typename OpTy, typename MatchContext>
+  bool match(OpTy *V, const MatchContext &Ctx) {
+    assert(Ctx.getTLI() && "TargetLowering is required for this pattern");
+    if (auto *I = dyn_cast<BinaryOperator>(V))
+      return (PredFunc(*Ctx.getTLI()) &&
+              ((L.match(I->getOperand(0)) && R.match(I->getOperand(1))) ||
+               (Commutable && L.match(I->getOperand(1)) &&
+                R.match(I->getOperand(0)))));
+    return false;
+  }
+};
+
+template <typename LHS, typename RHS>
+inline auto m_AnyBinOp(const LHS &L, const RHS &R) {
+  return AnyBinaryOp_match{[](const TargetLowering &TLI) { return true; }, L, R,
----------------
mshockwave wrote:

> Can you elaborate what you mean by:
> 
> > Furthermore, m_AnyBinOp(unsigned &Opc, LHS, RHS) is basically extracting the opcode that fulfills m_AnyBinOp(LHS, RHS).

So `m_AnyBinOp(LHS, RHS)` would check if the SDNode's opcode fulfills TLI.isBinOP/isCommutativeBinOp; `m_AnyBinOp(unsigned &Opc, LHS, RHS)` is basically doing the same thing `m_AnyBinOp(LHS, RHS)` does, plus returning the opcode (that has been checked) via the `Opc` argument (as `Opc` has a type of unsigned reference, making it an output argument).

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


More information about the llvm-commits mailing list