[embedded] Fix compilation crash with -num-threads and -mergeable-symbols

Author: kubamracek

lib/IRGen/IRGen.cpp

@@ -1666,7 +1666,8 @@ static void performParallelIRGeneration(IRGenDescriptor desc) {
       llvm::Module *M = IGM->getModule();
       auto collectReference = [&](llvm::GlobalValue &G) {
         if (G.isDeclaration()
-            && (G.getLinkage() == GlobalValue::LinkOnceODRLinkage ||
+            && (G.getLinkage() == GlobalValue::WeakODRLinkage ||
+                G.getLinkage() == GlobalValue::LinkOnceODRLinkage ||
                 G.getLinkage() == GlobalValue::ExternalLinkage)) {
           referencedGlobals.insert(G.getName());
           G.setLinkage(GlobalValue::ExternalLinkage);
@@ -1693,7 +1694,8 @@ static void performParallelIRGeneration(IRGenDescriptor desc) {
       // definition (if it's not referenced in the same file).
       auto updateLinkage = [&](llvm::GlobalValue &G) {
         if (!G.isDeclaration()
-            && G.getLinkage() == GlobalValue::LinkOnceODRLinkage
+            && (G.getLinkage() == GlobalValue::WeakODRLinkage ||
+                G.getLinkage() == GlobalValue::LinkOnceODRLinkage)
             && referencedGlobals.count(G.getName()) != 0) {
           G.setLinkage(GlobalValue::WeakODRLinkage);
         }

test/embedded/linkage-mergeable3.swift

@@ -0,0 +1,35 @@
+// RUN: %empty-directory(%t)
+// RUN: %{python} %utils/split_file.py -o %t %s
+
+// RUN: %target-swift-frontend -mergeable-symbols -num-threads 2 -O -c -emit-module -o %t/MyModule.o             %t/MyModule.swift                   -enable-experimental-feature Embedded -parse-as-library
+// RUN: %target-swift-frontend -mergeable-symbols -num-threads 2 -O -c              -o %t/MainA.o -o %t/MainB.o  %t/MainA.swift %t/MainB.swift -I %t -enable-experimental-feature Embedded -parse-as-library
+// RUN: %target-clang %t/MainA.o %t/MainB.o %t/MyModule.o -o %t/a.out
+// RUN: %target-run %t/a.out | %FileCheck %s
+
+// REQUIRES: swift_in_compiler
+// REQUIRES: executable_test
+// REQUIRES: swift_feature_Embedded
+
+// BEGIN MyModule.swift
+
+public func module_func(n: Int) {
+  var a: [Int] = [1, 2, 3]
+  print("module_func: \(a[0])")
+}
+
+// BEGIN MainA.swift
+
+import MyModule
+
+// BEGIN MainB.swift
+
+import MyModule
+
+@main
+struct Main {
+  static func main() {
+    module_func(n: 5)
+  }
+}
+
+// CHECK: module_func: 1

test/embedded/linkage-mergeable4.swift

@@ -0,0 +1,51 @@
+// RUN: %empty-directory(%t)
+// RUN: %{python} %utils/split_file.py -o %t %s
+
+// RUN: %target-swift-frontend -mergeable-symbols -num-threads 2 -O -c -emit-module -o %t/MyModule.o             %t/MyModule.swift                   -enable-experimental-feature Embedded -parse-as-library
+// RUN: %target-swift-frontend -mergeable-symbols -num-threads 2 -O -c              -o %t/MainA.o -o %t/MainB.o  %t/MainA.swift %t/MainB.swift -I %t -enable-experimental-feature Embedded -parse-as-library
+// RUN: %target-clang %t/MainA.o %t/MainB.o %t/MyModule.o -o %t/a.out
+// RUN: %target-run %t/a.out | %FileCheck %s
+
+// REQUIRES: swift_in_compiler
+// REQUIRES: executable_test
+// REQUIRES: swift_feature_Embedded
+
+// BEGIN MyModule.swift
+
+var dict: [Int:Int] = [1: 2, 3: 4]
+
+public func module_func() {
+  print("module_func: \(dict[1]!)")
+}
+
+// BEGIN MainA.swift
+
+import MyModule
+
+fileprivate var dict: [Int:Int] = [5: 6, 7: 8]
+
+func mainA_func() {
+  print("main_func: \(dict[5]!)")
+}
+
+// BEGIN MainB.swift
+
+import MyModule
+
+fileprivate var dict: [Int:Int] = [5: 6, 7: 8]
+
+func mainB_func() {
+  print("main_func: \(dict[5]!)")
+}
+
+@main
+struct Main {
+  static func main() {
+    module_func()
+    mainA_func()
+    mainB_func()
+  }
+}
+
+// CHECK: module_func: 2
+// CHECK: main_func: 6