Store TheCU

Author: arsenm

llvm/lib/CodeGen/AsmPrinter/CodeViewDebug.cpp

@@ -632,19 +632,19 @@ void CodeViewDebug::beginModule(Module *M) {
       Asm = nullptr;
       return;
     }
-
     Node = *CUs->operands().begin();
   }
-  const auto *CU = cast<DICompileUnit>(Node);
-  DISourceLanguageName Lang = CU->getSourceLanguage();
+
+  TheCU = cast<DICompileUnit>(Node);
+  DISourceLanguageName Lang = TheCU->getSourceLanguage();
   CurrentSourceLanguage =
       Lang.hasVersionedName()
           ? MapDWARFLanguageToCVLang(
                 static_cast<dwarf::SourceLanguageName>(Lang.getName()))
           : MapDWARFLanguageToCVLang(
                 static_cast<dwarf::SourceLanguage>(Lang.getName()));
   if (!M->getCodeViewFlag() ||
-      CU->getEmissionKind() == DICompileUnit::NoDebug) {
+      TheCU->getEmissionKind() == DICompileUnit::NoDebug) {
     Asm = nullptr;
     return;
   }
@@ -905,14 +905,9 @@ void CodeViewDebug::emitCompilerInformation() {
   OS.AddComment("CPUType");
   OS.emitInt16(static_cast<uint64_t>(TheCPU));
 
-  NamedMDNode *CUs = MMI->getModule()->getNamedMetadata("llvm.dbg.cu");
-
   StringRef CompilerVersion = "0";
-  if (!CUs->operands().empty()) {
-    const MDNode *Node = *CUs->operands().begin();
-    const auto *CU = cast<DICompileUnit>(Node);
-    CompilerVersion = CU->getProducer();
-  }
+  if (TheCU)
+    CompilerVersion = TheCU->getProducer();
 
   Version FrontVer = parseVersion(CompilerVersion);
   OS.AddComment("Frontend version");

llvm/lib/CodeGen/AsmPrinter/CodeViewDebug.h

@@ -98,6 +98,8 @@ class LLVM_LIBRARY_VISIBILITY CodeViewDebug : public DebugHandlerBase {
   /// The codeview CPU type used by the translation unit.
   codeview::CPUType TheCPU;
 
+  const DICompileUnit *TheCU = nullptr;
+
   /// The AsmPrinter used for emitting compiler metadata. When only compiler
   /// info is being emitted, DebugHandlerBase::Asm may be null.
   AsmPrinter *CompilerInfoAsm = nullptr;