Fixed missing uses of OwningMachineModuleInfoWrapperPass.

Author: matinraayai

llvm/unittests/CodeGen/AsmPrinterDwarfTest.cpp

@@ -403,7 +403,7 @@ class AsmPrinterHandlerTest : public AsmPrinterFixtureBase {
     AP->addAsmPrinterHandler(std::make_unique<TestHandler>(*this));
     TargetMachine *TM = &AP->TM;
     legacy::PassManager PM;
-    PM.add(new MachineModuleInfoWrapperPass(TM));
+    PM.add(new OwningMachineModuleInfoWrapperPass(*TM));
     PM.add(TestPrinter->releaseAP()); // Takes ownership of destroying AP
     LLVMContext Context;
     std::unique_ptr<Module> M(new Module("TestModule", Context));

llvm/unittests/CodeGen/X86MCInstLowerTest.cpp

@@ -149,7 +149,7 @@ class X86MCInstLowerTest : public testing::Test {
 TEST_F(X86MCInstLowerTest, moExternalSymbol_MCSYMBOL) {
 
   MachineModuleInfoWrapperPass *MMIWP =
-      new MachineModuleInfoWrapperPass(TM.get(), &*MCFoo);
+      new OwningMachineModuleInfoWrapperPass(*TM, *MCFoo);
 
   SmallString<1024> Buf;
   llvm::raw_svector_ostream OS(Buf);

llvm/unittests/MI/LiveIntervalTest.cpp

@@ -70,7 +70,8 @@ std::unique_ptr<Module> parseMIR(LLVMContext &Context,
 
   M->setDataLayout(TM.createDataLayout());
 
-  MachineModuleInfoWrapperPass *MMIWP = new MachineModuleInfoWrapperPass(&TM);
+  MachineModuleInfoWrapperPass *MMIWP =
+      new OwningMachineModuleInfoWrapperPass(TM);
   if (MIR->parseMachineFunctions(*M, MMIWP->getMMI()))
     return nullptr;
   PM.add(MMIWP);