[ThinLTO] Don't convert functions to declarations if force-import-all is enabled

On one hand, we intend to `force-import-all` functions when the option is
enabled. On the other hand, we currently drop definitions of some functions and
convert them to declarations, which contradicts this intent.

With this PR, functions will no longer be converted to declarations when
`force-import-all` is enabled.

Author: shiltian

llvm/lib/Transforms/IPO/FunctionImport.cpp

@@ -1701,13 +1701,16 @@ void llvm::thinLTOFinalizeInModule(Module &TheModule,
     if (NewLinkage == GV.getLinkage())
       return;
 
+    bool ForceImportFunction = isa<Function>(GV) && ForceImportAll;
+
     // Check for a non-prevailing def that has interposable linkage
     // (e.g. non-odr weak or linkonce). In that case we can't simply
     // convert to available_externally, since it would lose the
     // interposable property and possibly get inlined. Simply drop
     // the definition in that case.
     if (GlobalValue::isAvailableExternallyLinkage(NewLinkage) &&
-        GlobalValue::isInterposableLinkage(GV.getLinkage())) {
+        GlobalValue::isInterposableLinkage(GV.getLinkage()) &&
+        !ForceImportFunction) {
       if (!convertToDeclaration(GV))
         // FIXME: Change this to collect replaced GVs and later erase
         // them from the parent module once thinLTOResolvePrevailingGUID is

llvm/test/ThinLTO/AMDGPU/Inputs/in-0.ll

@@ -0,0 +1,13 @@
+define void @f1(ptr %v) #0 {
+entry:
+  call void @weak_common(ptr %v)
+  ret void
+}
+
+define weak hidden void @weak_common(ptr %v) #0 {
+entry:
+  store i32 12345, ptr %v
+  ret void
+}
+
+attributes #0 = { noinline }

llvm/test/ThinLTO/AMDGPU/Inputs/in-1.ll

@@ -0,0 +1,13 @@
+define void @f2(ptr %v) #0 {
+entry:
+  call void @weak_common(ptr %v)
+  ret void
+}
+
+define weak hidden void @weak_common(ptr %v) #0 {
+entry:
+  store i32 12345, ptr %v
+  ret void
+}
+
+attributes #0 = { noinline }

llvm/test/ThinLTO/AMDGPU/force-import-all.ll

@@ -0,0 +1,35 @@
+; RUN: opt -mtriple=amdgcn-amd-amdhsa -module-summary %s -o %t.main.bc
+; RUN: opt -mtriple=amdgcn-amd-amdhsa -module-summary %p/Inputs/in-0.ll -o %t.in.0.bc
+; RUN: opt -mtriple=amdgcn-amd-amdhsa -module-summary %p/Inputs/in-1.ll -o %t.in.1.bc
+; RUN: llvm-lto -thinlto-action=run -force-import-all %t.main.bc %t.in.0.bc %t.in.1.bc --thinlto-save-temps=%t.2.
+; RUN: llvm-dis %t.2.0.3.imported.bc -o - | FileCheck --check-prefix=MOD1 %s
+; RUN: llvm-dis %t.2.1.3.imported.bc -o - | FileCheck --check-prefix=MOD2 %s
+; RUN: llvm-dis %t.2.2.3.imported.bc -o - | FileCheck --check-prefix=MOD3 %s
+
+define void @f0(ptr %p) #0 {
+entry:
+  call void @f1(ptr %p)
+  call void @f2(ptr %p)
+  ret void
+}
+
+define weak hidden void @weak_common(ptr %v) #0 {
+entry:
+  store i32 12345, ptr %v
+  ret void
+}
+
+declare void @f1(ptr)
+
+declare void @f2(ptr)
+
+attributes #0 = { noinline }
+
+; MOD1: define available_externally void @f1
+; MOD1: define available_externally void @f2
+
+; MOD2: define void @f1
+; MOD2: define available_externally hidden void @weak_common
+
+; MOD3: define void @f2
+; MOD3: define available_externally hidden void @weak_common

llvm/test/ThinLTO/AMDGPU/lit.local.cfg

@@ -0,0 +1,2 @@
+if not "AMDGPU" in config.root.targets:
+    config.unsupported = True