Move loop to AsmPrinter

Author: redstar

llvm/lib/CodeGen/AsmPrinter/AsmPrinter.cpp

@@ -2937,6 +2937,21 @@ bool AsmPrinter::doFinalization(Module &M) {
     }
   }
 
+  // Emit symbol attributes for declarations (GOFF only).
+  if (TM.getTargetTriple().isOSBinFormatGOFF()) {
+    for (auto &GO : M.global_objects()) {
+      if (GO.isDeclaration()) {
+        MCSymbol *Sym = TM.getSymbol(&GO);
+        OutStreamer->emitSymbolAttribute(Sym, GO.hasExternalWeakLinkage()
+                                                  ? MCSA_WeakReference
+                                                  : MCSA_Global);
+        OutStreamer->emitSymbolAttribute(Sym, isa<Function>(GO)
+                                                  ? MCSA_ELF_TypeFunction
+                                                  : MCSA_ELF_TypeObject);
+      }
+    }
+  }
+
   // Allow the target to emit any magic that it wants at the end of the file,
   // after everything else has gone out.
   emitEndOfAsmFile(M);

llvm/lib/Target/SystemZ/SystemZAsmPrinter.cpp

@@ -1113,18 +1113,6 @@ void SystemZAsmPrinter::emitEndOfAsmFile(Module &M) {
   if (TT.isOSzOS()) {
     emitADASection();
     emitIDRLSection(M);
-    // Emit EXTRN declarations.
-    for (auto &GO : M.global_objects()) {
-      if (GO.isDeclaration()) {
-        MCSymbol *Sym = TM.getSymbol(&GO);
-        OutStreamer->emitSymbolAttribute(Sym, GO.hasExternalWeakLinkage()
-                                                  ? MCSA_WeakReference
-                                                  : MCSA_Global);
-        OutStreamer->emitSymbolAttribute(Sym, isa<Function>(GO)
-                                                  ? MCSA_ELF_TypeFunction
-                                                  : MCSA_ELF_TypeObject);
-      }
-    }
   }
   emitAttributes(M);
 }