Fixed compilation issue.

Author: matinraayai

llvm/examples/Kaleidoscope/Chapter8/toy.cpp

@@ -1,5 +1,6 @@
 #include "llvm/ADT/APFloat.h"
 #include "llvm/ADT/STLExtras.h"
+#include "llvm/CodeGen/MachineModuleInfo.h"
 #include "llvm/IR/BasicBlock.h"
 #include "llvm/IR/Constants.h"
 #include "llvm/IR/DerivedTypes.h"
@@ -1257,9 +1258,12 @@ int main() {
   }
 
   legacy::PassManager pass;
+  llvm::MachineModuleInfo MMI(
+      static_cast<LLVMTargetMachine *>(TheTargetMachine));
   auto FileType = CodeGenFileType::ObjectFile;
 
-  if (TheTargetMachine->addPassesToEmitFile(pass, dest, nullptr, FileType)) {
+  if (TheTargetMachine->addPassesToEmitFile(pass, MMI, dest, nullptr,
+                                            FileType)) {
     errs() << "TheTargetMachine can't emit a file of this type";
     return 1;
   }

llvm/include/llvm/Target/TargetMachine.h

@@ -34,7 +34,7 @@ using ModulePassManager = PassManager<Module>;
 
 class Function;
 class GlobalValue;
-class MachineModuleInfoWrapperPass;
+class MachineModuleInfo;
 class Mangler;
 class MCAsmInfo;
 class MCContext;

llvm/unittests/MI/LiveIntervalTest.cpp

@@ -209,7 +209,7 @@ static void doTest(StringRef MIRFunc,
     return;
 
   legacy::PassManager PM;
-  MachineModuleInfo MMI(TM->get());
+  MachineModuleInfo MMI(TM.get());
   std::unique_ptr<MIRParser> MIR;
   std::unique_ptr<Module> M = parseMIR(Context, MMI, MIR, *TM, MIRFunc);
   ASSERT_TRUE(M);

mlir/lib/Target/LLVM/ModuleToObject.cpp

@@ -20,6 +20,7 @@
 #include "mlir/Target/LLVMIR/ModuleTranslation.h"
 
 #include "llvm/Bitcode/BitcodeWriter.h"
+#include "llvm/CodeGen/MachineModuleInfo.h"
 #include "llvm/IR/LegacyPassManager.h"
 #include "llvm/IRReader/IRReader.h"
 #include "llvm/Linker/Linker.h"
@@ -175,8 +176,11 @@ ModuleToObject::translateToISA(llvm::Module &llvmModule,
   { // Drop pstream after this to prevent the ISA from being stuck buffering
     llvm::buffer_ostream pstream(stream);
     llvm::legacy::PassManager codegenPasses;
+    llvm::MachineModuleInfo machineModuleInfo(
+        static_cast<llvm::LLVMTargetMachine *>(&targetMachine));
 
-    if (targetMachine.addPassesToEmitFile(codegenPasses, pstream, nullptr,
+    if (targetMachine.addPassesToEmitFile(codegenPasses, machineModuleInfo,
+                                          pstream, nullptr,
                                           llvm::CodeGenFileType::AssemblyFile))
       return std::nullopt;