PatternMatchTest.cpp 18 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518
  1. //===- PatternMatchTest.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. #include "llvm/CodeGen/GlobalISel/ConstantFoldingMIRBuilder.h"
  9. #include "llvm/CodeGen/GlobalISel/MIPatternMatch.h"
  10. #include "llvm/CodeGen/GlobalISel/MachineIRBuilder.h"
  11. #include "llvm/CodeGen/GlobalISel/Utils.h"
  12. #include "llvm/CodeGen/MIRParser/MIRParser.h"
  13. #include "llvm/CodeGen/MachineFunction.h"
  14. #include "llvm/CodeGen/MachineModuleInfo.h"
  15. #include "llvm/CodeGen/TargetFrameLowering.h"
  16. #include "llvm/CodeGen/TargetInstrInfo.h"
  17. #include "llvm/CodeGen/TargetLowering.h"
  18. #include "llvm/CodeGen/TargetSubtargetInfo.h"
  19. #include "llvm/Support/SourceMgr.h"
  20. #include "llvm/Support/TargetRegistry.h"
  21. #include "llvm/Support/TargetSelect.h"
  22. #include "llvm/Target/TargetMachine.h"
  23. #include "llvm/Target/TargetOptions.h"
  24. #include "gtest/gtest.h"
  25. using namespace llvm;
  26. using namespace MIPatternMatch;
  27. namespace {
  28. void initLLVM() {
  29. InitializeAllTargets();
  30. InitializeAllTargetMCs();
  31. InitializeAllAsmPrinters();
  32. InitializeAllAsmParsers();
  33. PassRegistry *Registry = PassRegistry::getPassRegistry();
  34. initializeCore(*Registry);
  35. initializeCodeGen(*Registry);
  36. }
  37. /// Create a TargetMachine. As we lack a dedicated always available target for
  38. /// unittests, we go for "AArch64".
  39. std::unique_ptr<LLVMTargetMachine> createTargetMachine() {
  40. Triple TargetTriple("aarch64--");
  41. std::string Error;
  42. const Target *T = TargetRegistry::lookupTarget("", TargetTriple, Error);
  43. if (!T)
  44. return nullptr;
  45. TargetOptions Options;
  46. return std::unique_ptr<LLVMTargetMachine>(static_cast<LLVMTargetMachine*>(
  47. T->createTargetMachine("AArch64", "", "", Options, None, None,
  48. CodeGenOpt::Aggressive)));
  49. }
  50. std::unique_ptr<Module> parseMIR(LLVMContext &Context,
  51. std::unique_ptr<MIRParser> &MIR,
  52. const TargetMachine &TM, StringRef MIRCode,
  53. const char *FuncName, MachineModuleInfo &MMI) {
  54. SMDiagnostic Diagnostic;
  55. std::unique_ptr<MemoryBuffer> MBuffer = MemoryBuffer::getMemBuffer(MIRCode);
  56. MIR = createMIRParser(std::move(MBuffer), Context);
  57. if (!MIR)
  58. return nullptr;
  59. std::unique_ptr<Module> M = MIR->parseIRModule();
  60. if (!M)
  61. return nullptr;
  62. M->setDataLayout(TM.createDataLayout());
  63. if (MIR->parseMachineFunctions(*M, MMI))
  64. return nullptr;
  65. return M;
  66. }
  67. std::pair<std::unique_ptr<Module>, std::unique_ptr<MachineModuleInfo>>
  68. createDummyModule(LLVMContext &Context, const LLVMTargetMachine &TM,
  69. StringRef MIRFunc) {
  70. SmallString<512> S;
  71. StringRef MIRString = (Twine(R"MIR(
  72. ---
  73. ...
  74. name: func
  75. registers:
  76. - { id: 0, class: _ }
  77. - { id: 1, class: _ }
  78. - { id: 2, class: _ }
  79. - { id: 3, class: _ }
  80. body: |
  81. bb.1:
  82. %0(s64) = COPY $x0
  83. %1(s64) = COPY $x1
  84. %2(s64) = COPY $x2
  85. )MIR") + Twine(MIRFunc) + Twine("...\n"))
  86. .toNullTerminatedStringRef(S);
  87. std::unique_ptr<MIRParser> MIR;
  88. auto MMI = make_unique<MachineModuleInfo>(&TM);
  89. std::unique_ptr<Module> M =
  90. parseMIR(Context, MIR, TM, MIRString, "func", *MMI);
  91. return make_pair(std::move(M), std::move(MMI));
  92. }
  93. static MachineFunction *getMFFromMMI(const Module *M,
  94. const MachineModuleInfo *MMI) {
  95. Function *F = M->getFunction("func");
  96. auto *MF = MMI->getMachineFunction(*F);
  97. return MF;
  98. }
  99. static void collectCopies(SmallVectorImpl<unsigned> &Copies,
  100. MachineFunction *MF) {
  101. for (auto &MBB : *MF)
  102. for (MachineInstr &MI : MBB) {
  103. if (MI.getOpcode() == TargetOpcode::COPY)
  104. Copies.push_back(MI.getOperand(0).getReg());
  105. }
  106. }
  107. TEST(PatternMatchInstr, MatchIntConstant) {
  108. LLVMContext Context;
  109. std::unique_ptr<LLVMTargetMachine> TM = createTargetMachine();
  110. if (!TM)
  111. return;
  112. auto ModuleMMIPair = createDummyModule(Context, *TM, "");
  113. MachineFunction *MF =
  114. getMFFromMMI(ModuleMMIPair.first.get(), ModuleMMIPair.second.get());
  115. SmallVector<unsigned, 4> Copies;
  116. collectCopies(Copies, MF);
  117. MachineBasicBlock *EntryMBB = &*MF->begin();
  118. MachineIRBuilder B(*MF);
  119. MachineRegisterInfo &MRI = MF->getRegInfo();
  120. B.setInsertPt(*EntryMBB, EntryMBB->end());
  121. auto MIBCst = B.buildConstant(LLT::scalar(64), 42);
  122. int64_t Cst;
  123. bool match = mi_match(MIBCst->getOperand(0).getReg(), MRI, m_ICst(Cst));
  124. EXPECT_TRUE(match);
  125. EXPECT_EQ(Cst, 42);
  126. }
  127. TEST(PatternMatchInstr, MatchBinaryOp) {
  128. LLVMContext Context;
  129. std::unique_ptr<LLVMTargetMachine> TM = createTargetMachine();
  130. if (!TM)
  131. return;
  132. auto ModuleMMIPair = createDummyModule(Context, *TM, "");
  133. MachineFunction *MF =
  134. getMFFromMMI(ModuleMMIPair.first.get(), ModuleMMIPair.second.get());
  135. SmallVector<unsigned, 4> Copies;
  136. collectCopies(Copies, MF);
  137. MachineBasicBlock *EntryMBB = &*MF->begin();
  138. MachineIRBuilder B(*MF);
  139. MachineRegisterInfo &MRI = MF->getRegInfo();
  140. B.setInsertPt(*EntryMBB, EntryMBB->end());
  141. LLT s64 = LLT::scalar(64);
  142. auto MIBAdd = B.buildAdd(s64, Copies[0], Copies[1]);
  143. // Test case for no bind.
  144. bool match =
  145. mi_match(MIBAdd->getOperand(0).getReg(), MRI, m_GAdd(m_Reg(), m_Reg()));
  146. EXPECT_TRUE(match);
  147. Register Src0, Src1, Src2;
  148. match = mi_match(MIBAdd->getOperand(0).getReg(), MRI,
  149. m_GAdd(m_Reg(Src0), m_Reg(Src1)));
  150. EXPECT_TRUE(match);
  151. EXPECT_EQ(Src0, Copies[0]);
  152. EXPECT_EQ(Src1, Copies[1]);
  153. // Build MUL(ADD %0, %1), %2
  154. auto MIBMul = B.buildMul(s64, MIBAdd, Copies[2]);
  155. // Try to match MUL.
  156. match = mi_match(MIBMul->getOperand(0).getReg(), MRI,
  157. m_GMul(m_Reg(Src0), m_Reg(Src1)));
  158. EXPECT_TRUE(match);
  159. EXPECT_EQ(Src0, MIBAdd->getOperand(0).getReg());
  160. EXPECT_EQ(Src1, Copies[2]);
  161. // Try to match MUL(ADD)
  162. match = mi_match(MIBMul->getOperand(0).getReg(), MRI,
  163. m_GMul(m_GAdd(m_Reg(Src0), m_Reg(Src1)), m_Reg(Src2)));
  164. EXPECT_TRUE(match);
  165. EXPECT_EQ(Src0, Copies[0]);
  166. EXPECT_EQ(Src1, Copies[1]);
  167. EXPECT_EQ(Src2, Copies[2]);
  168. // Test Commutativity.
  169. auto MIBMul2 = B.buildMul(s64, Copies[0], B.buildConstant(s64, 42));
  170. // Try to match MUL(Cst, Reg) on src of MUL(Reg, Cst) to validate
  171. // commutativity.
  172. int64_t Cst;
  173. match = mi_match(MIBMul2->getOperand(0).getReg(), MRI,
  174. m_GMul(m_ICst(Cst), m_Reg(Src0)));
  175. EXPECT_TRUE(match);
  176. EXPECT_EQ(Cst, 42);
  177. EXPECT_EQ(Src0, Copies[0]);
  178. // Make sure commutative doesn't work with something like SUB.
  179. auto MIBSub = B.buildSub(s64, Copies[0], B.buildConstant(s64, 42));
  180. match = mi_match(MIBSub->getOperand(0).getReg(), MRI,
  181. m_GSub(m_ICst(Cst), m_Reg(Src0)));
  182. EXPECT_FALSE(match);
  183. auto MIBFMul = B.buildInstr(TargetOpcode::G_FMUL, {s64},
  184. {Copies[0], B.buildConstant(s64, 42)});
  185. // Match and test commutativity for FMUL.
  186. match = mi_match(MIBFMul->getOperand(0).getReg(), MRI,
  187. m_GFMul(m_ICst(Cst), m_Reg(Src0)));
  188. EXPECT_TRUE(match);
  189. EXPECT_EQ(Cst, 42);
  190. EXPECT_EQ(Src0, Copies[0]);
  191. // FSUB
  192. auto MIBFSub = B.buildInstr(TargetOpcode::G_FSUB, {s64},
  193. {Copies[0], B.buildConstant(s64, 42)});
  194. match = mi_match(MIBFSub->getOperand(0).getReg(), MRI,
  195. m_GFSub(m_Reg(Src0), m_Reg()));
  196. EXPECT_TRUE(match);
  197. EXPECT_EQ(Src0, Copies[0]);
  198. // Build AND %0, %1
  199. auto MIBAnd = B.buildAnd(s64, Copies[0], Copies[1]);
  200. // Try to match AND.
  201. match = mi_match(MIBAnd->getOperand(0).getReg(), MRI,
  202. m_GAnd(m_Reg(Src0), m_Reg(Src1)));
  203. EXPECT_TRUE(match);
  204. EXPECT_EQ(Src0, Copies[0]);
  205. EXPECT_EQ(Src1, Copies[1]);
  206. // Build OR %0, %1
  207. auto MIBOr = B.buildOr(s64, Copies[0], Copies[1]);
  208. // Try to match OR.
  209. match = mi_match(MIBOr->getOperand(0).getReg(), MRI,
  210. m_GOr(m_Reg(Src0), m_Reg(Src1)));
  211. EXPECT_TRUE(match);
  212. EXPECT_EQ(Src0, Copies[0]);
  213. EXPECT_EQ(Src1, Copies[1]);
  214. // Try to use the FoldableInstructionsBuilder to build binary ops.
  215. ConstantFoldingMIRBuilder CFB(B.getState());
  216. LLT s32 = LLT::scalar(32);
  217. auto MIBCAdd =
  218. CFB.buildAdd(s32, CFB.buildConstant(s32, 0), CFB.buildConstant(s32, 1));
  219. // This should be a constant now.
  220. match = mi_match(MIBCAdd->getOperand(0).getReg(), MRI, m_ICst(Cst));
  221. EXPECT_TRUE(match);
  222. EXPECT_EQ(Cst, 1);
  223. auto MIBCAdd1 =
  224. CFB.buildInstr(TargetOpcode::G_ADD, {s32},
  225. {CFB.buildConstant(s32, 0), CFB.buildConstant(s32, 1)});
  226. // This should be a constant now.
  227. match = mi_match(MIBCAdd1->getOperand(0).getReg(), MRI, m_ICst(Cst));
  228. EXPECT_TRUE(match);
  229. EXPECT_EQ(Cst, 1);
  230. // Try one of the other constructors of MachineIRBuilder to make sure it's
  231. // compatible.
  232. ConstantFoldingMIRBuilder CFB1(*MF);
  233. CFB1.setInsertPt(*EntryMBB, EntryMBB->end());
  234. auto MIBCSub =
  235. CFB1.buildInstr(TargetOpcode::G_SUB, {s32},
  236. {CFB1.buildConstant(s32, 1), CFB1.buildConstant(s32, 1)});
  237. // This should be a constant now.
  238. match = mi_match(MIBCSub->getOperand(0).getReg(), MRI, m_ICst(Cst));
  239. EXPECT_TRUE(match);
  240. EXPECT_EQ(Cst, 0);
  241. }
  242. TEST(PatternMatchInstr, MatchFPUnaryOp) {
  243. LLVMContext Context;
  244. std::unique_ptr<LLVMTargetMachine> TM = createTargetMachine();
  245. if (!TM)
  246. return;
  247. auto ModuleMMIPair = createDummyModule(Context, *TM, "");
  248. MachineFunction *MF =
  249. getMFFromMMI(ModuleMMIPair.first.get(), ModuleMMIPair.second.get());
  250. SmallVector<unsigned, 4> Copies;
  251. collectCopies(Copies, MF);
  252. MachineBasicBlock *EntryMBB = &*MF->begin();
  253. MachineIRBuilder B(*MF);
  254. MachineRegisterInfo &MRI = MF->getRegInfo();
  255. B.setInsertPt(*EntryMBB, EntryMBB->end());
  256. // Truncate s64 to s32.
  257. LLT s32 = LLT::scalar(32);
  258. auto Copy0s32 = B.buildFPTrunc(s32, Copies[0]);
  259. // Match G_FABS.
  260. auto MIBFabs = B.buildInstr(TargetOpcode::G_FABS, {s32}, {Copy0s32});
  261. bool match = mi_match(MIBFabs->getOperand(0).getReg(), MRI, m_GFabs(m_Reg()));
  262. EXPECT_TRUE(match);
  263. Register Src;
  264. auto MIBFNeg = B.buildInstr(TargetOpcode::G_FNEG, {s32}, {Copy0s32});
  265. match = mi_match(MIBFNeg->getOperand(0).getReg(), MRI, m_GFNeg(m_Reg(Src)));
  266. EXPECT_TRUE(match);
  267. EXPECT_EQ(Src, Copy0s32->getOperand(0).getReg());
  268. match = mi_match(MIBFabs->getOperand(0).getReg(), MRI, m_GFabs(m_Reg(Src)));
  269. EXPECT_TRUE(match);
  270. EXPECT_EQ(Src, Copy0s32->getOperand(0).getReg());
  271. // Build and match FConstant.
  272. auto MIBFCst = B.buildFConstant(s32, .5);
  273. const ConstantFP *TmpFP{};
  274. match = mi_match(MIBFCst->getOperand(0).getReg(), MRI, m_GFCst(TmpFP));
  275. EXPECT_TRUE(match);
  276. EXPECT_TRUE(TmpFP);
  277. APFloat APF((float).5);
  278. auto *CFP = ConstantFP::get(Context, APF);
  279. EXPECT_EQ(CFP, TmpFP);
  280. // Build double float.
  281. LLT s64 = LLT::scalar(64);
  282. auto MIBFCst64 = B.buildFConstant(s64, .5);
  283. const ConstantFP *TmpFP64{};
  284. match = mi_match(MIBFCst64->getOperand(0).getReg(), MRI, m_GFCst(TmpFP64));
  285. EXPECT_TRUE(match);
  286. EXPECT_TRUE(TmpFP64);
  287. APFloat APF64(.5);
  288. auto CFP64 = ConstantFP::get(Context, APF64);
  289. EXPECT_EQ(CFP64, TmpFP64);
  290. EXPECT_NE(TmpFP64, TmpFP);
  291. // Build half float.
  292. LLT s16 = LLT::scalar(16);
  293. auto MIBFCst16 = B.buildFConstant(s16, .5);
  294. const ConstantFP *TmpFP16{};
  295. match = mi_match(MIBFCst16->getOperand(0).getReg(), MRI, m_GFCst(TmpFP16));
  296. EXPECT_TRUE(match);
  297. EXPECT_TRUE(TmpFP16);
  298. bool Ignored;
  299. APFloat APF16(.5);
  300. APF16.convert(APFloat::IEEEhalf(), APFloat::rmNearestTiesToEven, &Ignored);
  301. auto CFP16 = ConstantFP::get(Context, APF16);
  302. EXPECT_EQ(TmpFP16, CFP16);
  303. EXPECT_NE(TmpFP16, TmpFP);
  304. }
  305. TEST(PatternMatchInstr, MatchExtendsTrunc) {
  306. LLVMContext Context;
  307. std::unique_ptr<LLVMTargetMachine> TM = createTargetMachine();
  308. if (!TM)
  309. return;
  310. auto ModuleMMIPair = createDummyModule(Context, *TM, "");
  311. MachineFunction *MF =
  312. getMFFromMMI(ModuleMMIPair.first.get(), ModuleMMIPair.second.get());
  313. SmallVector<unsigned, 4> Copies;
  314. collectCopies(Copies, MF);
  315. MachineBasicBlock *EntryMBB = &*MF->begin();
  316. MachineIRBuilder B(*MF);
  317. MachineRegisterInfo &MRI = MF->getRegInfo();
  318. B.setInsertPt(*EntryMBB, EntryMBB->end());
  319. LLT s64 = LLT::scalar(64);
  320. LLT s32 = LLT::scalar(32);
  321. auto MIBTrunc = B.buildTrunc(s32, Copies[0]);
  322. auto MIBAExt = B.buildAnyExt(s64, MIBTrunc);
  323. auto MIBZExt = B.buildZExt(s64, MIBTrunc);
  324. auto MIBSExt = B.buildSExt(s64, MIBTrunc);
  325. Register Src0;
  326. bool match =
  327. mi_match(MIBTrunc->getOperand(0).getReg(), MRI, m_GTrunc(m_Reg(Src0)));
  328. EXPECT_TRUE(match);
  329. EXPECT_EQ(Src0, Copies[0]);
  330. match =
  331. mi_match(MIBAExt->getOperand(0).getReg(), MRI, m_GAnyExt(m_Reg(Src0)));
  332. EXPECT_TRUE(match);
  333. EXPECT_EQ(Src0, MIBTrunc->getOperand(0).getReg());
  334. match = mi_match(MIBSExt->getOperand(0).getReg(), MRI, m_GSExt(m_Reg(Src0)));
  335. EXPECT_TRUE(match);
  336. EXPECT_EQ(Src0, MIBTrunc->getOperand(0).getReg());
  337. match = mi_match(MIBZExt->getOperand(0).getReg(), MRI, m_GZExt(m_Reg(Src0)));
  338. EXPECT_TRUE(match);
  339. EXPECT_EQ(Src0, MIBTrunc->getOperand(0).getReg());
  340. // Match ext(trunc src)
  341. match = mi_match(MIBAExt->getOperand(0).getReg(), MRI,
  342. m_GAnyExt(m_GTrunc(m_Reg(Src0))));
  343. EXPECT_TRUE(match);
  344. EXPECT_EQ(Src0, Copies[0]);
  345. match = mi_match(MIBSExt->getOperand(0).getReg(), MRI,
  346. m_GSExt(m_GTrunc(m_Reg(Src0))));
  347. EXPECT_TRUE(match);
  348. EXPECT_EQ(Src0, Copies[0]);
  349. match = mi_match(MIBZExt->getOperand(0).getReg(), MRI,
  350. m_GZExt(m_GTrunc(m_Reg(Src0))));
  351. EXPECT_TRUE(match);
  352. EXPECT_EQ(Src0, Copies[0]);
  353. }
  354. TEST(PatternMatchInstr, MatchSpecificType) {
  355. LLVMContext Context;
  356. std::unique_ptr<LLVMTargetMachine> TM = createTargetMachine();
  357. if (!TM)
  358. return;
  359. auto ModuleMMIPair = createDummyModule(Context, *TM, "");
  360. MachineFunction *MF =
  361. getMFFromMMI(ModuleMMIPair.first.get(), ModuleMMIPair.second.get());
  362. SmallVector<unsigned, 4> Copies;
  363. collectCopies(Copies, MF);
  364. MachineBasicBlock *EntryMBB = &*MF->begin();
  365. MachineIRBuilder B(*MF);
  366. MachineRegisterInfo &MRI = MF->getRegInfo();
  367. B.setInsertPt(*EntryMBB, EntryMBB->end());
  368. // Try to match a 64bit add.
  369. LLT s64 = LLT::scalar(64);
  370. LLT s32 = LLT::scalar(32);
  371. auto MIBAdd = B.buildAdd(s64, Copies[0], Copies[1]);
  372. EXPECT_FALSE(mi_match(MIBAdd->getOperand(0).getReg(), MRI,
  373. m_GAdd(m_SpecificType(s32), m_Reg())));
  374. EXPECT_TRUE(mi_match(MIBAdd->getOperand(0).getReg(), MRI,
  375. m_GAdd(m_SpecificType(s64), m_Reg())));
  376. // Try to match the destination type of a bitcast.
  377. LLT v2s32 = LLT::vector(2, 32);
  378. auto MIBCast = B.buildCast(v2s32, Copies[0]);
  379. EXPECT_TRUE(
  380. mi_match(MIBCast->getOperand(0).getReg(), MRI, m_GBitcast(m_Reg())));
  381. EXPECT_TRUE(
  382. mi_match(MIBCast->getOperand(0).getReg(), MRI, m_SpecificType(v2s32)));
  383. EXPECT_TRUE(
  384. mi_match(MIBCast->getOperand(1).getReg(), MRI, m_SpecificType(s64)));
  385. // Build a PTRToInt and INTTOPTR and match and test them.
  386. LLT PtrTy = LLT::pointer(0, 64);
  387. auto MIBIntToPtr = B.buildCast(PtrTy, Copies[0]);
  388. auto MIBPtrToInt = B.buildCast(s64, MIBIntToPtr);
  389. Register Src0;
  390. // match the ptrtoint(inttoptr reg)
  391. bool match = mi_match(MIBPtrToInt->getOperand(0).getReg(), MRI,
  392. m_GPtrToInt(m_GIntToPtr(m_Reg(Src0))));
  393. EXPECT_TRUE(match);
  394. EXPECT_EQ(Src0, Copies[0]);
  395. }
  396. TEST(PatternMatchInstr, MatchCombinators) {
  397. LLVMContext Context;
  398. std::unique_ptr<LLVMTargetMachine> TM = createTargetMachine();
  399. if (!TM)
  400. return;
  401. auto ModuleMMIPair = createDummyModule(Context, *TM, "");
  402. MachineFunction *MF =
  403. getMFFromMMI(ModuleMMIPair.first.get(), ModuleMMIPair.second.get());
  404. SmallVector<unsigned, 4> Copies;
  405. collectCopies(Copies, MF);
  406. MachineBasicBlock *EntryMBB = &*MF->begin();
  407. MachineIRBuilder B(*MF);
  408. MachineRegisterInfo &MRI = MF->getRegInfo();
  409. B.setInsertPt(*EntryMBB, EntryMBB->end());
  410. LLT s64 = LLT::scalar(64);
  411. LLT s32 = LLT::scalar(32);
  412. auto MIBAdd = B.buildAdd(s64, Copies[0], Copies[1]);
  413. Register Src0, Src1;
  414. bool match =
  415. mi_match(MIBAdd->getOperand(0).getReg(), MRI,
  416. m_all_of(m_SpecificType(s64), m_GAdd(m_Reg(Src0), m_Reg(Src1))));
  417. EXPECT_TRUE(match);
  418. EXPECT_EQ(Src0, Copies[0]);
  419. EXPECT_EQ(Src1, Copies[1]);
  420. // Check for s32 (which should fail).
  421. match =
  422. mi_match(MIBAdd->getOperand(0).getReg(), MRI,
  423. m_all_of(m_SpecificType(s32), m_GAdd(m_Reg(Src0), m_Reg(Src1))));
  424. EXPECT_FALSE(match);
  425. match =
  426. mi_match(MIBAdd->getOperand(0).getReg(), MRI,
  427. m_any_of(m_SpecificType(s32), m_GAdd(m_Reg(Src0), m_Reg(Src1))));
  428. EXPECT_TRUE(match);
  429. EXPECT_EQ(Src0, Copies[0]);
  430. EXPECT_EQ(Src1, Copies[1]);
  431. // Match a case where none of the predicates hold true.
  432. match = mi_match(
  433. MIBAdd->getOperand(0).getReg(), MRI,
  434. m_any_of(m_SpecificType(LLT::scalar(16)), m_GSub(m_Reg(), m_Reg())));
  435. EXPECT_FALSE(match);
  436. }
  437. TEST(PatternMatchInstr, MatchMiscellaneous) {
  438. LLVMContext Context;
  439. std::unique_ptr<LLVMTargetMachine> TM = createTargetMachine();
  440. if (!TM)
  441. return;
  442. auto ModuleMMIPair = createDummyModule(Context, *TM, "");
  443. MachineFunction *MF =
  444. getMFFromMMI(ModuleMMIPair.first.get(), ModuleMMIPair.second.get());
  445. SmallVector<unsigned, 4> Copies;
  446. collectCopies(Copies, MF);
  447. MachineBasicBlock *EntryMBB = &*MF->begin();
  448. MachineIRBuilder B(*MF);
  449. MachineRegisterInfo &MRI = MF->getRegInfo();
  450. B.setInsertPt(*EntryMBB, EntryMBB->end());
  451. LLT s64 = LLT::scalar(64);
  452. auto MIBAdd = B.buildAdd(s64, Copies[0], Copies[1]);
  453. // Make multiple uses of this add.
  454. B.buildCast(LLT::pointer(0, 32), MIBAdd);
  455. B.buildCast(LLT::pointer(1, 32), MIBAdd);
  456. bool match = mi_match(MIBAdd.getReg(0), MRI, m_GAdd(m_Reg(), m_Reg()));
  457. EXPECT_TRUE(match);
  458. match = mi_match(MIBAdd.getReg(0), MRI, m_OneUse(m_GAdd(m_Reg(), m_Reg())));
  459. EXPECT_FALSE(match);
  460. }
  461. } // namespace
  462. int main(int argc, char **argv) {
  463. ::testing::InitGoogleTest(&argc, argv);
  464. initLLVM();
  465. return RUN_ALL_TESTS();
  466. }