[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