averello/swift-mirror
1
0
Fork 0
mirror of https://github.com/apple/swift.git synced 2025-12-21 12:14:44 +01:00
Code Issues Packages Projects Releases Wiki Activity

[region-isolation] Add mocking for getSourceInst so that unittests still work.

Browse Source
This commit is contained in:
Michael Gottesman
2024-09-04 12:55:04 -07:00
parent 3cfcc173e9
commit eda603d2c9
2 changed files with 19 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
include/swift/SILOptimizer/Utils/PartitionUtils.h
View File

@@ -1229,11 +1229,14 @@ public:
// case, if we are not marked explicitly as sending, we do not transfer
// since the disconnected value is allowed to be resued after we
// return. If we are passed as a sending parameter, we cannot do this.
if (auto fas = FullApplySite::isa(op.getSourceInst());
(!fas || !fas.isSending(*op.getSourceOp())) &&
transferredRegionIsolation.isDisconnected() && calleeIsolationInfo &&
transferredRegionIsolation.hasSameIsolation(calleeIsolationInfo))
return;
if (auto *sourceInst = Impl::getSourceInst(op)) {
if (auto fas = FullApplySite::isa(sourceInst);
(!fas || !fas.isSending(*op.getSourceOp())) &&
transferredRegionIsolation.isDisconnected() &&
calleeIsolationInfo &&
transferredRegionIsolation.hasSameIsolation(calleeIsolationInfo))
return;
}
// Mark op.getOpArgs()[0] as transferred.
TransferringOperandState &state = operandToStateMap.get(op.getSourceOp());
@@ -1581,6 +1584,9 @@ struct PartitionOpEvaluatorBaseImpl : PartitionOpEvaluator<Subclass> {
static SILLocation getLoc(SILInstruction *inst) { return inst->getLoc(); }
static SILLocation getLoc(Operand *op) { return op->getUser()->getLoc(); }
static SILInstruction *getSourceInst(const PartitionOp &partitionOp) {
return partitionOp.getSourceInst();
}
static SILIsolationInfo getIsolationInfo(const PartitionOp &partitionOp) {
return SILIsolationInfo::get(partitionOp.getSourceInst());
}