[Distributed] Fix type lowering for Builtin.initializeDistributedRemoteActor

Author: drexin

lib/IRGen/GenBuiltin.cpp

@@ -390,7 +390,7 @@ void irgen::emitBuiltinCall(IRGenFunction &IGF, const BuiltinInfo &Builtin,
 
   if (Builtin.ID == BuiltinValueKind::InitializeDistributedRemoteActor) {
     auto actorMetatype = args.claimNext();
-    emitDistributedActorInitializeRemote(IGF, actorMetatype, out);
+    emitDistributedActorInitializeRemote(IGF, resultType, actorMetatype, out);
     return;
   }
 

lib/IRGen/GenDistributed.cpp

@@ -17,6 +17,7 @@
 #include "GenDistributed.h"
 
 #include "BitPatternBuilder.h"
+#include "ClassTypeInfo.h"
 #include "ExtraInhabitants.h"
 #include "GenProto.h"
 #include "GenType.h"
@@ -32,7 +33,12 @@ using namespace swift;
 using namespace irgen;
 
 llvm::Value *irgen::emitDistributedActorInitializeRemote(
-    IRGenFunction &IGF, llvm::Value *actorMetatype, Explosion &out) {
+    IRGenFunction &IGF, SILType selfType, llvm::Value *actorMetatype, Explosion &out) {
+  auto &classTI = IGF.getTypeInfo(selfType).as<ClassTypeInfo>();
+  auto &classLayout = classTI.getClassLayout(IGF.IGM, selfType,
+                                             /*forBackwardDeployment=*/false);
+  llvm::Type *destType = classLayout.getType()->getPointerTo();
+
   auto fn = IGF.IGM.getDistributedActorInitializeRemoteFn();
   actorMetatype =
       IGF.Builder.CreateBitCast(actorMetatype, IGF.IGM.TypeMetadataPtrTy);
@@ -41,9 +47,11 @@ llvm::Value *irgen::emitDistributedActorInitializeRemote(
   call->setCallingConv(IGF.IGM.SwiftCC);
   call->setDoesNotThrow();
 
-  out.add(call);
+  auto result = IGF.Builder.CreateBitCast(call, destType);
+
+  out.add(result);
 
-  return call;
+  return result;
 }
 
 void irgen::emitDistributedActorDestroy(IRGenFunction &IGF,

lib/IRGen/GenDistributed.h

@@ -43,6 +43,7 @@ class IRGenFunction;
 /// Emit the '_distributedActorRemoteInitialize' call.
 llvm::Value *emitDistributedActorInitializeRemote(
     IRGenFunction &IGF,
+    SILType selfType,
     llvm::Value *actorMetatype,
     Explosion &out);