support large code model

Author: Wael Yehia

llvm/lib/Target/PowerPC/PPCAsmPrinter.cpp

@@ -3576,18 +3576,35 @@ void PPCAIXAsmPrinter::emitGlobalIFunc(Module &M, const GlobalIFunc &GI) {
     return;
   }
 
-  //  lwz 12, L..foo_desc(2)
   auto FnDescTOCEntryType = getTOCEntryTypeForLinkage(GI.getLinkage());
-  auto *FnDescTOCEntrySym =
-      lookUpOrCreateTOCEntry(CurrentFnDescSym, FnDescTOCEntryType);
-  auto *Exp = MCSymbolRefExpr::create(FnDescTOCEntrySym, OutContext);
-  // Exp = getTOCEntryLoadingExprForXCOFF(MOSymbol, Exp, VK);// TODO: need this?
-  // need this uncommented
-  OutStreamer->emitInstruction(MCInstBuilder(IsPPC64 ? PPC::LD : PPC::LWZ)
-                                   .addReg(PPC::X12)
-                                   .addExpr(Exp)
-                                   .addReg(PPC::X2),
-                               *Subtarget);
+  auto *FnDescTOCEntrySym = lookUpOrCreateTOCEntry(CurrentFnDescSym, FnDescTOCEntryType);
+
+  if (TM.getCodeModel() == CodeModel::Large) {
+    //  addis 12, L..foo_desc@u(2)
+    //  lwz 12, L..foo_desc@l(12)
+    auto *Exp_U = symbolWithSpecifier(FnDescTOCEntrySym, PPC::S_U);
+    OutStreamer->emitInstruction(MCInstBuilder(PPC::ADDIS)
+                                         .addReg(PPC::X12)
+                                         .addReg(PPC::X2)
+                                         .addExpr(Exp_U),
+                                     *Subtarget);
+    auto *Exp_L = symbolWithSpecifier(FnDescTOCEntrySym, PPC::S_L);
+    OutStreamer->emitInstruction(MCInstBuilder(IsPPC64 ? PPC::LD : PPC::LWZ)
+                                             .addReg(PPC::X12)
+                                             .addExpr(Exp_L)
+                                             .addReg(PPC::X12),
+                                         *Subtarget);
+  } else {
+    //  lwz 12, L..foo_desc(2)
+    auto *Exp = MCSymbolRefExpr::create(FnDescTOCEntrySym, OutContext);
+    // Exp = getTOCEntryLoadingExprForXCOFF(MOSymbol, Exp, VK);
+    // TODO: do we need to uncomment this?
+    OutStreamer->emitInstruction(MCInstBuilder(IsPPC64 ? PPC::LD : PPC::LWZ)
+                                     .addReg(PPC::X12)
+                                     .addExpr(Exp)
+                                     .addReg(PPC::X2),
+                                 *Subtarget);
+  }
   //  lwz 11, 8(12)
   OutStreamer->emitInstruction(MCInstBuilder(IsPPC64 ? PPC::LD : PPC::LWZ)
                                    .addReg(PPC::X11)

llvm/test/CodeGen/PowerPC/aix-ifunc.ll

@@ -4,6 +4,9 @@
 ; RUN: llc -mtriple=powerpc64-ibm-aix-xcoff --function-sections %s -o - | FileCheck %s --check-prefixes=COMMON,FUNCSECT -DALIGN=3 -DPTR_SIZE=8 -DLOAD=ld -DOFF=16
 ; RUN: llc -mtriple=powerpc-ibm-aix-xcoff --function-sections %s -o - | FileCheck %s --check-prefixes=COMMON,FUNCSECT -DALIGN=2 -DPTR_SIZE=4 -DLOAD=lwz -DOFF=8
 
+; RUN: llc -mtriple=powerpc64-ibm-aix-xcoff --function-sections --code-model=large %s -o - | FileCheck %s --check-prefixes=LARGE -DALIGN=3 -DPTR_SIZE=8 -DLOAD=ld -DOFF=16
+; RUN: llc -mtriple=powerpc-ibm-aix-xcoff --function-sections --code-model=large %s -o - | FileCheck %s --check-prefixes=LARGE -DALIGN=2 -DPTR_SIZE=4 -DLOAD=lwz -DOFF=8
+
 ;;;; section __ifunc_sec holding the [foo:foo_resolver] pairs
 ; COMMON:              .csect __ifunc_sec[RW],2
 ; COMMON-NEXT:         .align  [[ALIGN]]
@@ -46,10 +49,18 @@
 ; COMMON-NEXT:         mtctr 12
 ; COMMON-NEXT:         bctr
 
+; -mcmodel=large:
+; LARGE:             .csect .foo[PR],5
+; LARGE:             addis 12, [[FOO_TOC:.*]]@u(2)
+; LARGE-NEXT:        [[LOAD]] 12, [[FOO_TOC]]@l(12)
+; LARGE-NEXT:        [[LOAD]] 11, [[OFF]](12)
+; LARGE-NEXT:        [[LOAD]] 12, 0(12)
+
 ;;;; foo's TOC entry
 ; COMMON: [[FOO_TOC]]:
 ; COMMON-NEXT:         .tc foo[TC],foo[DS]
-
+; LARGE: [[FOO_TOC]]:
+; LARGE-NEXT:          .tc foo[TE],foo[DS]
 
 @foo = ifunc i32 (...), ptr @foo.resolver