Skip to content

Commit ef10cd0

Browse files
ktosoktoso
committed
[Distributed] Fix generic returning distributed funcs
1 parent 67c097a commit ef10cd0

File tree

5 files changed

+63
-29
lines changed

5 files changed

+63
-29
lines changed

‎lib/SIL/IR/SILFunction.cpp

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -319,6 +319,8 @@ SILType SILFunction::mapTypeIntoContext(SILType type) const {
319319

320320
SILType GenericEnvironment::mapTypeIntoContext(SILModule &M,
321321
SILType type) const {
322+
if (type.hasArchetype())
323+
type.dump();
322324
assert(!type.hasArchetype());
323325

324326
auto genericSig = getGenericSignature().getCanonicalSignature();

‎lib/SILGen/SILGenDistributed.cpp

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -118,7 +118,9 @@ static void emitDistributedIfRemoteBranch(SILGenFunction &SGF, SILLocation Loc,
118118
"'Distributed' module available?");
119119

120120
ManagedValue selfAnyObject = B.createInitExistentialRef(
121-
Loc, SGF.getLoweredType(ctx.getAnyObjectType()), CanType(selfTy),
121+
Loc,
122+
/*existentialType=*/SGF.getLoweredType(ctx.getAnyObjectType()),
123+
/*formalConcreteType=*/selfValue.getType().getASTType(),
122124
selfValue, {});
123125
auto result = SGF.emitApplyOfLibraryIntrinsic(
124126
Loc, isRemoteFn, SubstitutionMap(), {selfAnyObject}, SGFContext());
@@ -379,6 +381,7 @@ void SILGenFunction::emitDistributedActorFactory(FuncDecl *fd) { // TODO(distrib
379381
/// having at least one call to the resolve function.
380382

381383
auto &C = getASTContext();
384+
auto DC = fd->getDeclContext();
382385
SILLocation loc = fd;
383386

384387
// ==== Prepare argument references
@@ -392,12 +395,12 @@ void SILGenFunction::emitDistributedActorFactory(FuncDecl *fd) { // TODO(distrib
392395
ManagedValue selfArg = ManagedValue::forUnmanaged(selfArgValue);
393396

394397
// type: SpecificDistributedActor.Type
395-
auto selfArgType = F.mapTypeIntoContext(selfArg.getType().getASTType());
398+
auto selfArgType = selfArg.getType().getASTType();
396399
auto selfMetatype = getLoweredType(selfArgType);
397400
SILValue selfMetatypeValue = B.createMetatype(loc, selfMetatype);
398401

399402
// type: SpecificDistributedActor
400-
auto *selfTyDecl = fd->getParent()->getSelfNominalTypeDecl();
403+
auto *selfTyDecl = DC->getSelfClassDecl();
401404
assert(selfTyDecl->isDistributedActor());
402405
auto selfTy = F.mapTypeIntoContext(selfTyDecl->getDeclaredInterfaceType());
403406
auto returnTy = getLoweredType(selfTy);
@@ -533,7 +536,8 @@ SILGenFunction::emitConditionalResignIdentityCall(SILLocation loc,
533536
assert(actorDecl->isDistributedActor() &&
534537
"only distributed actors have actorSystem lifecycle hooks in deinit");
535538

536-
auto selfTy = actorDecl->getDeclaredInterfaceType();
539+
auto selfTy = F.mapTypeIntoContext(actorDecl->getDeclaredInterfaceType());
540+
fprintf(stderr, "[%s:%d] (%s) OKEY!!!!!\n", __FILE__, __LINE__, __FUNCTION__);
537541

538542
// we only system.resignID if we are a local actor,
539543
// and thus the address was created by system.assignID.

‎lib/SILOptimizer/Utils/DistributedActor.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -183,7 +183,7 @@ void emitActorReadyCall(SILBuilder &B, SILLocation loc, SILValue actor,
183183
auto &C = F.getASTContext();
184184
emitDistributedActorSystemWitnessCall(
185185
B, loc, C.Id_actorReady, actorSystem,
186-
F.mapTypeIntoContext(actor->getType()), { actor });
186+
actor->getType(), { actor });
187187
}
188188

189189
void emitResignIdentityCall(SILBuilder &B, SILLocation loc,

‎lib/Sema/CodeSynthesisDistributedActor.cpp

Lines changed: 24 additions & 24 deletions
Original file line numberDiff line numberDiff line change
@@ -55,7 +55,7 @@ static VarDecl *addImplicitDistributedActorIDProperty(
5555
// ==== Synthesize and add 'id' property to the actor decl
5656
Type propertyType = getDistributedActorIDType(nominal);
5757

58-
VarDecl *propDecl = new (C)
58+
auto *propDecl = new (C)
5959
VarDecl(/*IsStatic*/false, VarDecl::Introducer::Let,
6060
SourceLoc(), C.Id_id, nominal);
6161
propDecl->setImplicit();
@@ -230,20 +230,21 @@ deriveBodyDistributed_thunk(AbstractFunctionDecl *thunk, void *context) {
230230

231231
// --- Recording invocation details
232232
// -- recordGenericSubstitution(s)
233-
if (func->isGeneric() || nominal->isGeneric()) {
233+
if (thunk->isGeneric() || nominal->isGeneric()) {
234234
auto recordGenericSubstitutionDecl =
235235
C.getRecordGenericSubstitutionOnDistributedInvocationEncoder(invocationEncoderDecl);
236236
assert(recordGenericSubstitutionDecl);
237237
auto recordGenericSubstitutionDeclRef =
238238
UnresolvedDeclRefExpr::createImplicit(
239239
C, recordGenericSubstitutionDecl->getName());
240240

241-
auto sig = func->getGenericSignature();
242-
for (auto genParamType : sig.getGenericParams()) {
241+
auto signature = thunk->getGenericSignature();
242+
for (auto genParamType : signature.getGenericParams()) {
243243

244+
auto tyExpr = TypeExpr::createImplicit(thunk->mapTypeIntoContext(genParamType), C);
244245
auto subTypeExpr = new (C) DotSelfExpr(
245-
TypeExpr::createImplicit(thunk->mapTypeIntoContext(genParamType), C),
246-
sloc, sloc, thunk->mapTypeIntoContext(genParamType));
246+
tyExpr,
247+
sloc, sloc, tyExpr->getType());
247248

248249
auto recordGenericSubArgsList =
249250
ArgumentList::forImplicitCallTo(
@@ -378,7 +379,8 @@ deriveBodyDistributed_thunk(AbstractFunctionDecl *thunk, void *context) {
378379
// -- recordReturnType
379380
if (!isVoidReturn) {
380381
// Result.self
381-
auto resultType = func->getResultInterfaceType();
382+
// Watch out and always map into thunk context
383+
auto resultType = thunk->mapTypeIntoContext(func->getResultInterfaceType());
382384
auto *metaTypeRef = TypeExpr::createImplicit(resultType, C);
383385
auto *resultTypeExpr =
384386
new (C) DotSelfExpr(metaTypeRef, sloc, sloc, resultType);
@@ -426,9 +428,8 @@ deriveBodyDistributed_thunk(AbstractFunctionDecl *thunk, void *context) {
426428
}
427429

428430
// === Prepare the 'RemoteCallTarget'
429-
VarDecl *targetVar =
430-
new (C) VarDecl(/*isStatic=*/false, VarDecl::Introducer::Let, sloc,
431-
C.Id_target, thunk);
431+
auto *targetVar = new (C) VarDecl(
432+
/*isStatic=*/false, VarDecl::Introducer::Let, sloc, C.Id_target, thunk);
432433

433434
{
434435
// --- Mangle the thunk name
@@ -510,7 +511,8 @@ deriveBodyDistributed_thunk(AbstractFunctionDecl *thunk, void *context) {
510511
// -- returning: Res.Type
511512
if (!isVoidReturn) {
512513
// Result.self
513-
auto resultType = func->getResultInterfaceType();
514+
auto resultType =
515+
func->mapTypeIntoContext(func->getResultInterfaceType());
514516
auto *metaTypeRef = TypeExpr::createImplicit(resultType, C);
515517
auto *resultTypeExpr =
516518
new (C) DotSelfExpr(metaTypeRef, sloc, sloc, resultType);
@@ -561,7 +563,7 @@ static FuncDecl *createDistributedThunkFunction(FuncDecl *func) {
561563
genericParamList = genericParams->clone(DC);
562564
}
563565

564-
GenericSignature thunkGenSig =
566+
GenericSignature baseSignature =
565567
buildGenericSignature(C, func->getGenericSignature(),
566568
/*addedParameters=*/{},
567569
/*addedRequirements=*/{});
@@ -581,9 +583,9 @@ static FuncDecl *createDistributedThunkFunction(FuncDecl *func) {
581583
}
582584

583585
auto paramDecl = new (C)
584-
ParamDecl(SourceLoc(),
585-
/*argumentNameLoc=*/SourceLoc(), funcParam->getArgumentName(),
586-
/*parameterNameLoc=*/SourceLoc(), paramName, DC);
586+
ParamDecl(SourceLoc(),
587+
/*argumentNameLoc=*/SourceLoc(), funcParam->getArgumentName(),
588+
/*parameterNameLoc=*/SourceLoc(), paramName, DC);
587589

588590
paramDecl->setImplicit(true);
589591
paramDecl->setSpecifier(funcParam->getSpecifier());
@@ -593,16 +595,14 @@ static FuncDecl *createDistributedThunkFunction(FuncDecl *func) {
593595
}
594596
ParameterList *params = ParameterList::create(C, paramDecls); // = funcParams->clone(C);
595597

596-
auto thunk = FuncDecl::createImplicit(C, swift::StaticSpellingKind::None,
597-
thunkName, SourceLoc(),
598-
/*async=*/true, /*throws=*/true,
599-
genericParamList,
600-
params,
601-
func->getResultInterfaceType(),
602-
DC);
598+
auto thunk = FuncDecl::createImplicit(
599+
C, swift::StaticSpellingKind::None, thunkName, SourceLoc(),
600+
/*async=*/true, /*throws=*/true,
601+
genericParamList, params,
602+
func->getResultInterfaceType(), DC);
603603
thunk->setSynthesized(true);
604-
thunk->getAttrs().add(new (C) NonisolatedAttr(/*implicit=*/true));
605-
thunk->setGenericSignature(thunkGenSig);
604+
thunk->getAttrs().add(new (C) NonisolatedAttr(/*isImplicit=*/true));
605+
thunk->setGenericSignature(baseSignature);
606606
thunk->copyFormalAccessFrom(func, /*sourceIsParentContext=*/false);
607607
thunk->setBodySynthesizer(deriveBodyDistributed_thunk, func);
608608

‎test/Distributed/distributed_actor_generic_actor.swift

Lines changed: 28 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,28 @@
1+
// RUN: %empty-directory(%t)
2+
// RUN: %target-swift-frontend-emit-module -emit-module-path %t/FakeDistributedActorSystems.swiftmodule -module-name FakeDistributedActorSystems -disable-availability-checking %S/Inputs/FakeDistributedActorSystems.swift
3+
// RUN: %target-swift-frontend -typecheck -verify -enable-experimental-distributed -disable-availability-checking -I %t 2>&1 %s
4+
// REQUIRES: concurrency
5+
// REQUIRES: distributed
6+
7+
import Distributed
8+
import FakeDistributedActorSystems
9+
10+
distributed actor Worker<Work: Sendable & Codable> {
11+
typealias ActorSystem = FakeActorSystem
12+
13+
distributed func echo(item: Work) -> Work {
14+
item
15+
}
16+
17+
distributed func echo(items: [Work]) -> [Work] {
18+
items
19+
}
20+
21+
distributed func other<Other: Codable & Sendable>(other: Other) -> Other {
22+
other
23+
}
24+
25+
distributed func others<Other: Codable & Sendable>(other: Other) -> [Other] {
26+
[other]
27+
}
28+
}

0 commit comments

Comments
 (0)