xref: /llvm-project/llvm/unittests/CodeGen/SelectionDAGPatternMatchTest.cpp (revision 63e1647827f3427c5f3ad37461d84a63ba5fcdaf)
1 //===---- llvm/unittest/CodeGen/SelectionDAGPatternMatchTest.cpp  ---------===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 
9 #include "llvm/Analysis/OptimizationRemarkEmitter.h"
10 #include "llvm/AsmParser/Parser.h"
11 #include "llvm/CodeGen/MachineModuleInfo.h"
12 #include "llvm/CodeGen/SDPatternMatch.h"
13 #include "llvm/CodeGen/TargetLowering.h"
14 #include "llvm/IR/Module.h"
15 #include "llvm/MC/TargetRegistry.h"
16 #include "llvm/Support/SourceMgr.h"
17 #include "llvm/Support/TargetSelect.h"
18 #include "llvm/Target/TargetMachine.h"
19 #include "gtest/gtest.h"
20 
21 using namespace llvm;
22 
23 class SelectionDAGPatternMatchTest : public testing::Test {
24 protected:
25   static void SetUpTestCase() {
26     InitializeAllTargets();
27     InitializeAllTargetMCs();
28   }
29 
30   void SetUp() override {
31     StringRef Assembly = "@g = global i32 0\n"
32                          "@g_alias = alias i32, i32* @g\n"
33                          "define i32 @f() {\n"
34                          "  %1 = load i32, i32* @g\n"
35                          "  ret i32 %1\n"
36                          "}";
37 
38     Triple TargetTriple("riscv64--");
39     std::string Error;
40     const Target *T = TargetRegistry::lookupTarget("", TargetTriple, Error);
41     // FIXME: These tests do not depend on RISCV specifically, but we have to
42     // initialize a target. A skeleton Target for unittests would allow us to
43     // always run these tests.
44     if (!T)
45       GTEST_SKIP();
46 
47     TargetOptions Options;
48     TM = std::unique_ptr<LLVMTargetMachine>(static_cast<LLVMTargetMachine *>(
49         T->createTargetMachine("riscv64", "", "+m,+f,+d,+v", Options,
50                                std::nullopt, std::nullopt,
51                                CodeGenOptLevel::Aggressive)));
52     if (!TM)
53       GTEST_SKIP();
54 
55     SMDiagnostic SMError;
56     M = parseAssemblyString(Assembly, SMError, Context);
57     if (!M)
58       report_fatal_error(SMError.getMessage());
59     M->setDataLayout(TM->createDataLayout());
60 
61     F = M->getFunction("f");
62     if (!F)
63       report_fatal_error("F?");
64     G = M->getGlobalVariable("g");
65     if (!G)
66       report_fatal_error("G?");
67     AliasedG = M->getNamedAlias("g_alias");
68     if (!AliasedG)
69       report_fatal_error("AliasedG?");
70 
71     MachineModuleInfo MMI(TM.get());
72 
73     MF = std::make_unique<MachineFunction>(*F, *TM, *TM->getSubtargetImpl(*F),
74                                            MMI.getContext(), 0);
75 
76     DAG = std::make_unique<SelectionDAG>(*TM, CodeGenOptLevel::None);
77     if (!DAG)
78       report_fatal_error("DAG?");
79     OptimizationRemarkEmitter ORE(F);
80     DAG->init(*MF, ORE, nullptr, nullptr, nullptr, nullptr, nullptr, MMI,
81               nullptr);
82   }
83 
84   TargetLoweringBase::LegalizeTypeAction getTypeAction(EVT VT) {
85     return DAG->getTargetLoweringInfo().getTypeAction(Context, VT);
86   }
87 
88   EVT getTypeToTransformTo(EVT VT) {
89     return DAG->getTargetLoweringInfo().getTypeToTransformTo(Context, VT);
90   }
91 
92   LLVMContext Context;
93   std::unique_ptr<LLVMTargetMachine> TM;
94   std::unique_ptr<Module> M;
95   Function *F;
96   GlobalVariable *G;
97   GlobalAlias *AliasedG;
98   std::unique_ptr<MachineFunction> MF;
99   std::unique_ptr<SelectionDAG> DAG;
100 };
101 
102 TEST_F(SelectionDAGPatternMatchTest, matchValueType) {
103   SDLoc DL;
104   auto Int32VT = EVT::getIntegerVT(Context, 32);
105   auto Float32VT = EVT::getFloatingPointVT(32);
106   auto VInt32VT = EVT::getVectorVT(Context, Int32VT, 4);
107 
108   SDValue Op0 = DAG->getCopyFromReg(DAG->getEntryNode(), DL, 1, Int32VT);
109   SDValue Op1 = DAG->getCopyFromReg(DAG->getEntryNode(), DL, 2, Float32VT);
110   SDValue Op2 = DAG->getCopyFromReg(DAG->getEntryNode(), DL, 2, VInt32VT);
111 
112   using namespace SDPatternMatch;
113   EXPECT_TRUE(sd_match(Op0, m_SpecificVT(Int32VT)));
114   EVT BindVT;
115   EXPECT_TRUE(sd_match(Op1, m_VT(BindVT)));
116   EXPECT_EQ(BindVT, Float32VT);
117   EXPECT_TRUE(sd_match(Op0, m_IntegerVT()));
118   EXPECT_TRUE(sd_match(Op1, m_FloatingPointVT()));
119   EXPECT_TRUE(sd_match(Op2, m_VectorVT()));
120   EXPECT_FALSE(sd_match(Op2, m_ScalableVectorVT()));
121 }
122 
123 TEST_F(SelectionDAGPatternMatchTest, matchTernaryOp) {
124   SDLoc DL;
125   auto Int32VT = EVT::getIntegerVT(Context, 32);
126 
127   SDValue Op0 = DAG->getCopyFromReg(DAG->getEntryNode(), DL, 1, Int32VT);
128   SDValue Op1 = DAG->getCopyFromReg(DAG->getEntryNode(), DL, 2, Int32VT);
129 
130   SDValue ICMP_UGT = DAG->getSetCC(DL, MVT::i1, Op0, Op1, ISD::SETUGT);
131   SDValue ICMP_EQ01 = DAG->getSetCC(DL, MVT::i1, Op0, Op1, ISD::SETEQ);
132   SDValue ICMP_EQ10 = DAG->getSetCC(DL, MVT::i1, Op1, Op0, ISD::SETEQ);
133 
134   using namespace SDPatternMatch;
135   ISD::CondCode CC;
136   EXPECT_TRUE(sd_match(ICMP_UGT, m_SetCC(m_Value(), m_Value(),
137                                          m_SpecificCondCode(ISD::SETUGT))));
138   EXPECT_TRUE(
139       sd_match(ICMP_UGT, m_SetCC(m_Value(), m_Value(), m_CondCode(CC))));
140   EXPECT_TRUE(CC == ISD::SETUGT);
141   EXPECT_FALSE(sd_match(
142       ICMP_UGT, m_SetCC(m_Value(), m_Value(), m_SpecificCondCode(ISD::SETLE))));
143 
144   EXPECT_TRUE(sd_match(ICMP_EQ01, m_SetCC(m_Specific(Op0), m_Specific(Op1),
145                                           m_SpecificCondCode(ISD::SETEQ))));
146   EXPECT_TRUE(sd_match(ICMP_EQ10, m_SetCC(m_Specific(Op1), m_Specific(Op0),
147                                           m_SpecificCondCode(ISD::SETEQ))));
148   EXPECT_FALSE(sd_match(ICMP_EQ01, m_SetCC(m_Specific(Op1), m_Specific(Op0),
149                                            m_SpecificCondCode(ISD::SETEQ))));
150   EXPECT_FALSE(sd_match(ICMP_EQ10, m_SetCC(m_Specific(Op0), m_Specific(Op1),
151                                            m_SpecificCondCode(ISD::SETEQ))));
152   EXPECT_TRUE(sd_match(ICMP_EQ01, m_c_SetCC(m_Specific(Op1), m_Specific(Op0),
153                                             m_SpecificCondCode(ISD::SETEQ))));
154   EXPECT_TRUE(sd_match(ICMP_EQ10, m_c_SetCC(m_Specific(Op0), m_Specific(Op1),
155                                             m_SpecificCondCode(ISD::SETEQ))));
156 }
157 
158 TEST_F(SelectionDAGPatternMatchTest, matchBinaryOp) {
159   SDLoc DL;
160   auto Int32VT = EVT::getIntegerVT(Context, 32);
161   auto Float32VT = EVT::getFloatingPointVT(32);
162 
163   SDValue Op0 = DAG->getCopyFromReg(DAG->getEntryNode(), DL, 1, Int32VT);
164   SDValue Op1 = DAG->getCopyFromReg(DAG->getEntryNode(), DL, 2, Int32VT);
165   SDValue Op2 = DAG->getCopyFromReg(DAG->getEntryNode(), DL, 3, Float32VT);
166 
167   SDValue Add = DAG->getNode(ISD::ADD, DL, Int32VT, Op0, Op1);
168   SDValue Sub = DAG->getNode(ISD::SUB, DL, Int32VT, Add, Op0);
169   SDValue Mul = DAG->getNode(ISD::MUL, DL, Int32VT, Add, Sub);
170   SDValue And = DAG->getNode(ISD::AND, DL, Int32VT, Op0, Op1);
171   SDValue Xor = DAG->getNode(ISD::XOR, DL, Int32VT, Op1, Op0);
172   SDValue Or  = DAG->getNode(ISD::OR, DL, Int32VT, Op0, Op1);
173   SDValue SMax = DAG->getNode(ISD::SMAX, DL, Int32VT, Op0, Op1);
174   SDValue SMin = DAG->getNode(ISD::SMIN, DL, Int32VT, Op1, Op0);
175   SDValue UMax = DAG->getNode(ISD::UMAX, DL, Int32VT, Op0, Op1);
176   SDValue UMin = DAG->getNode(ISD::UMIN, DL, Int32VT, Op1, Op0);
177 
178   SDValue SFAdd = DAG->getNode(ISD::STRICT_FADD, DL, {Float32VT, MVT::Other},
179                                {DAG->getEntryNode(), Op2, Op2});
180 
181   using namespace SDPatternMatch;
182   EXPECT_TRUE(sd_match(Sub, m_BinOp(ISD::SUB, m_Value(), m_Value())));
183   EXPECT_TRUE(sd_match(Sub, m_Sub(m_Value(), m_Value())));
184   EXPECT_TRUE(sd_match(Add, m_c_BinOp(ISD::ADD, m_Value(), m_Value())));
185   EXPECT_TRUE(sd_match(Add, m_Add(m_Value(), m_Value())));
186   EXPECT_TRUE(sd_match(
187       Mul, m_Mul(m_OneUse(m_Opc(ISD::SUB)), m_NUses<2>(m_Specific(Add)))));
188   EXPECT_TRUE(
189       sd_match(SFAdd, m_ChainedBinOp(ISD::STRICT_FADD, m_SpecificVT(Float32VT),
190                                      m_SpecificVT(Float32VT))));
191 
192   EXPECT_TRUE(sd_match(And, m_c_BinOp(ISD::AND, m_Value(), m_Value())));
193   EXPECT_TRUE(sd_match(And, m_And(m_Value(), m_Value())));
194   EXPECT_TRUE(sd_match(Xor, m_c_BinOp(ISD::XOR, m_Value(), m_Value())));
195   EXPECT_TRUE(sd_match(Xor, m_Xor(m_Value(), m_Value())));
196   EXPECT_TRUE(sd_match(Or, m_c_BinOp(ISD::OR, m_Value(), m_Value())));
197   EXPECT_TRUE(sd_match(Or, m_Or(m_Value(), m_Value())));
198 
199   EXPECT_TRUE(sd_match(SMax, m_c_BinOp(ISD::SMAX, m_Value(), m_Value())));
200   EXPECT_TRUE(sd_match(SMax, m_SMax(m_Value(), m_Value())));
201   EXPECT_TRUE(sd_match(SMin, m_c_BinOp(ISD::SMIN, m_Value(), m_Value())));
202   EXPECT_TRUE(sd_match(SMin, m_SMin(m_Value(), m_Value())));
203   EXPECT_TRUE(sd_match(UMax, m_c_BinOp(ISD::UMAX, m_Value(), m_Value())));
204   EXPECT_TRUE(sd_match(UMax, m_UMax(m_Value(), m_Value())));
205   EXPECT_TRUE(sd_match(UMin, m_c_BinOp(ISD::UMIN, m_Value(), m_Value())));
206   EXPECT_TRUE(sd_match(UMin, m_UMin(m_Value(), m_Value())));
207 
208   SDValue BindVal;
209   EXPECT_TRUE(sd_match(SFAdd, m_ChainedBinOp(ISD::STRICT_FADD, m_Value(BindVal),
210                                              m_Deferred(BindVal))));
211   EXPECT_FALSE(sd_match(SFAdd, m_ChainedBinOp(ISD::STRICT_FADD, m_OtherVT(),
212                                               m_SpecificVT(Float32VT))));
213 }
214 
215 TEST_F(SelectionDAGPatternMatchTest, matchUnaryOp) {
216   SDLoc DL;
217   auto Int32VT = EVT::getIntegerVT(Context, 32);
218   auto Int64VT = EVT::getIntegerVT(Context, 64);
219 
220   SDValue Op0 = DAG->getCopyFromReg(DAG->getEntryNode(), DL, 1, Int32VT);
221   SDValue Op1 = DAG->getCopyFromReg(DAG->getEntryNode(), DL, 1, Int64VT);
222 
223   SDValue ZExt = DAG->getNode(ISD::ZERO_EXTEND, DL, Int64VT, Op0);
224   SDValue SExt = DAG->getNode(ISD::SIGN_EXTEND, DL, Int64VT, Op0);
225   SDValue Trunc = DAG->getNode(ISD::TRUNCATE, DL, Int32VT, Op1);
226 
227   SDValue Sub = DAG->getNode(ISD::SUB, DL, Int32VT, Trunc, Op0);
228   SDValue Neg = DAG->getNegative(Op0, DL, Int32VT);
229   SDValue Not = DAG->getNOT(DL, Op0, Int32VT);
230 
231   using namespace SDPatternMatch;
232   EXPECT_TRUE(sd_match(ZExt, m_UnaryOp(ISD::ZERO_EXTEND, m_Value())));
233   EXPECT_TRUE(sd_match(SExt, m_SExt(m_Value())));
234   EXPECT_TRUE(sd_match(Trunc, m_Trunc(m_Specific(Op1))));
235 
236   EXPECT_TRUE(sd_match(Neg, m_Neg(m_Value())));
237   EXPECT_TRUE(sd_match(Not, m_Not(m_Value())));
238   EXPECT_FALSE(sd_match(ZExt, m_Neg(m_Value())));
239   EXPECT_FALSE(sd_match(Sub, m_Neg(m_Value())));
240   EXPECT_FALSE(sd_match(Neg, m_Not(m_Value())));
241 }
242 
243 TEST_F(SelectionDAGPatternMatchTest, matchConstants) {
244   SDLoc DL;
245   auto Int32VT = EVT::getIntegerVT(Context, 32);
246   auto VInt32VT = EVT::getVectorVT(Context, Int32VT, 4);
247 
248   SDValue Arg0 = DAG->getCopyFromReg(DAG->getEntryNode(), DL, 1, Int32VT);
249 
250   SDValue Const3 = DAG->getConstant(3, DL, Int32VT);
251   SDValue Const87 = DAG->getConstant(87, DL, Int32VT);
252   SDValue Splat = DAG->getSplat(VInt32VT, DL, Arg0);
253   SDValue ConstSplat = DAG->getSplat(VInt32VT, DL, Const3);
254   SDValue Zero = DAG->getConstant(0, DL, Int32VT);
255   SDValue One = DAG->getConstant(1, DL, Int32VT);
256   SDValue AllOnes = DAG->getConstant(APInt::getAllOnes(32), DL, Int32VT);
257   SDValue SetCC = DAG->getSetCC(DL, Int32VT, Arg0, Const3, ISD::SETULT);
258 
259   using namespace SDPatternMatch;
260   EXPECT_TRUE(sd_match(Const87, m_ConstInt()));
261   EXPECT_FALSE(sd_match(Arg0, m_ConstInt()));
262   APInt ConstVal;
263   EXPECT_TRUE(sd_match(ConstSplat, m_ConstInt(ConstVal)));
264   EXPECT_EQ(ConstVal, 3);
265   EXPECT_FALSE(sd_match(Splat, m_ConstInt()));
266 
267   EXPECT_TRUE(sd_match(Const87, m_SpecificInt(87)));
268   EXPECT_TRUE(sd_match(Const3, m_SpecificInt(ConstVal)));
269   EXPECT_TRUE(sd_match(AllOnes, m_AllOnes()));
270 
271   EXPECT_TRUE(sd_match(Zero, DAG.get(), m_False()));
272   EXPECT_TRUE(sd_match(One, DAG.get(), m_True()));
273   EXPECT_FALSE(sd_match(AllOnes, DAG.get(), m_True()));
274 
275   ISD::CondCode CC;
276   EXPECT_TRUE(sd_match(
277       SetCC, m_Node(ISD::SETCC, m_Value(), m_Value(), m_CondCode(CC))));
278   EXPECT_EQ(CC, ISD::SETULT);
279   EXPECT_TRUE(sd_match(SetCC, m_Node(ISD::SETCC, m_Value(), m_Value(),
280                                      m_SpecificCondCode(ISD::SETULT))));
281 }
282 
283 TEST_F(SelectionDAGPatternMatchTest, patternCombinators) {
284   SDLoc DL;
285   auto Int32VT = EVT::getIntegerVT(Context, 32);
286 
287   SDValue Op0 = DAG->getCopyFromReg(DAG->getEntryNode(), DL, 1, Int32VT);
288   SDValue Op1 = DAG->getCopyFromReg(DAG->getEntryNode(), DL, 2, Int32VT);
289 
290   SDValue Add = DAG->getNode(ISD::ADD, DL, Int32VT, Op0, Op1);
291   SDValue Sub = DAG->getNode(ISD::SUB, DL, Int32VT, Add, Op0);
292 
293   using namespace SDPatternMatch;
294   EXPECT_TRUE(sd_match(
295       Sub, m_AnyOf(m_Opc(ISD::ADD), m_Opc(ISD::SUB), m_Opc(ISD::MUL))));
296   EXPECT_TRUE(sd_match(Add, m_AllOf(m_Opc(ISD::ADD), m_OneUse())));
297   EXPECT_TRUE(sd_match(Add, m_NoneOf(m_Opc(ISD::SUB), m_Opc(ISD::MUL))));
298 }
299 
300 TEST_F(SelectionDAGPatternMatchTest, optionalResizing) {
301   SDLoc DL;
302   auto Int32VT = EVT::getIntegerVT(Context, 32);
303   auto Int64VT = EVT::getIntegerVT(Context, 64);
304 
305   SDValue Op32 = DAG->getCopyFromReg(DAG->getEntryNode(), DL, 1, Int32VT);
306   SDValue Op64 = DAG->getCopyFromReg(DAG->getEntryNode(), DL, 1, Int64VT);
307   SDValue ZExt = DAG->getNode(ISD::ZERO_EXTEND, DL, Int64VT, Op32);
308   SDValue SExt = DAG->getNode(ISD::SIGN_EXTEND, DL, Int64VT, Op32);
309   SDValue AExt = DAG->getNode(ISD::ANY_EXTEND, DL, Int64VT, Op32);
310   SDValue Trunc = DAG->getNode(ISD::TRUNCATE, DL, Int32VT, Op64);
311 
312   using namespace SDPatternMatch;
313   SDValue A;
314   EXPECT_TRUE(sd_match(Op32, m_ZExtOrSelf(m_Value(A))));
315   EXPECT_TRUE(A == Op32);
316   EXPECT_TRUE(sd_match(ZExt, m_ZExtOrSelf(m_Value(A))));
317   EXPECT_TRUE(A == Op32);
318   EXPECT_TRUE(sd_match(Op64, m_SExtOrSelf(m_Value(A))));
319   EXPECT_TRUE(A == Op64);
320   EXPECT_TRUE(sd_match(SExt, m_SExtOrSelf(m_Value(A))));
321   EXPECT_TRUE(A == Op32);
322   EXPECT_TRUE(sd_match(Op32, m_AExtOrSelf(m_Value(A))));
323   EXPECT_TRUE(A == Op32);
324   EXPECT_TRUE(sd_match(AExt, m_AExtOrSelf(m_Value(A))));
325   EXPECT_TRUE(A == Op32);
326   EXPECT_TRUE(sd_match(Op64, m_TruncOrSelf(m_Value(A))));
327   EXPECT_TRUE(A == Op64);
328   EXPECT_TRUE(sd_match(Trunc, m_TruncOrSelf(m_Value(A))));
329   EXPECT_TRUE(A == Op64);
330 }
331 
332 TEST_F(SelectionDAGPatternMatchTest, matchNode) {
333   SDLoc DL;
334   auto Int32VT = EVT::getIntegerVT(Context, 32);
335 
336   SDValue Op0 = DAG->getCopyFromReg(DAG->getEntryNode(), DL, 1, Int32VT);
337   SDValue Op1 = DAG->getCopyFromReg(DAG->getEntryNode(), DL, 2, Int32VT);
338 
339   SDValue Add = DAG->getNode(ISD::ADD, DL, Int32VT, Op0, Op1);
340 
341   using namespace SDPatternMatch;
342   EXPECT_TRUE(sd_match(Add, m_Node(ISD::ADD, m_Value(), m_Value())));
343   EXPECT_FALSE(sd_match(Add, m_Node(ISD::SUB, m_Value(), m_Value())));
344   EXPECT_FALSE(sd_match(Add, m_Node(ISD::ADD, m_Value())));
345   EXPECT_FALSE(
346       sd_match(Add, m_Node(ISD::ADD, m_Value(), m_Value(), m_Value())));
347   EXPECT_FALSE(sd_match(Add, m_Node(ISD::ADD, m_ConstInt(), m_Value())));
348 }
349 
350 namespace {
351 struct VPMatchContext : public SDPatternMatch::BasicMatchContext {
352   using SDPatternMatch::BasicMatchContext::BasicMatchContext;
353 
354   bool match(SDValue OpVal, unsigned Opc) const {
355     if (!OpVal->isVPOpcode())
356       return OpVal->getOpcode() == Opc;
357 
358     auto BaseOpc = ISD::getBaseOpcodeForVP(OpVal->getOpcode(), false);
359     return BaseOpc.has_value() && *BaseOpc == Opc;
360   }
361 };
362 } // anonymous namespace
363 TEST_F(SelectionDAGPatternMatchTest, matchContext) {
364   SDLoc DL;
365   auto BoolVT = EVT::getIntegerVT(Context, 1);
366   auto Int32VT = EVT::getIntegerVT(Context, 32);
367   auto VInt32VT = EVT::getVectorVT(Context, Int32VT, 4);
368   auto MaskVT = EVT::getVectorVT(Context, BoolVT, 4);
369 
370   SDValue Scalar0 = DAG->getCopyFromReg(DAG->getEntryNode(), DL, 1, Int32VT);
371   SDValue Vector0 = DAG->getCopyFromReg(DAG->getEntryNode(), DL, 2, VInt32VT);
372   SDValue Mask0 = DAG->getCopyFromReg(DAG->getEntryNode(), DL, 3, MaskVT);
373 
374   SDValue VPAdd = DAG->getNode(ISD::VP_ADD, DL, VInt32VT,
375                                {Vector0, Vector0, Mask0, Scalar0});
376   SDValue VPReduceAdd = DAG->getNode(ISD::VP_REDUCE_ADD, DL, Int32VT,
377                                      {Scalar0, VPAdd, Mask0, Scalar0});
378 
379   using namespace SDPatternMatch;
380   VPMatchContext VPCtx(DAG.get());
381   EXPECT_TRUE(sd_context_match(VPAdd, VPCtx, m_Opc(ISD::ADD)));
382   // VP_REDUCE_ADD doesn't have a based opcode, so we use a normal
383   // sd_match before switching to VPMatchContext when checking VPAdd.
384   EXPECT_TRUE(sd_match(VPReduceAdd, m_Node(ISD::VP_REDUCE_ADD, m_Value(),
385                                            m_Context(VPCtx, m_Opc(ISD::ADD)),
386                                            m_Value(), m_Value())));
387 }
388 
389 TEST_F(SelectionDAGPatternMatchTest, matchAdvancedProperties) {
390   SDLoc DL;
391   auto Int16VT = EVT::getIntegerVT(Context, 16);
392   auto Int64VT = EVT::getIntegerVT(Context, 64);
393 
394   SDValue Op0 = DAG->getCopyFromReg(DAG->getEntryNode(), DL, 1, Int64VT);
395   SDValue Op1 = DAG->getCopyFromReg(DAG->getEntryNode(), DL, 2, Int16VT);
396 
397   SDValue Add = DAG->getNode(ISD::ADD, DL, Int64VT, Op0, Op0);
398 
399   using namespace SDPatternMatch;
400   EXPECT_TRUE(sd_match(Op0, DAG.get(), m_LegalType(m_Value())));
401   EXPECT_FALSE(sd_match(Op1, DAG.get(), m_LegalType(m_Value())));
402   EXPECT_TRUE(sd_match(Add, DAG.get(),
403                        m_LegalOp(m_IntegerVT(m_Add(m_Value(), m_Value())))));
404 }
405