Fix OwnershipKind of block argument while creating a preheader

meg-gupta · meg-gupta · commit 1c05e768b9b2 · 2022-12-13T21:24:07.000-08:00
diff --git a/lib/SILOptimizer/Utils/LoopUtils.cpp b/lib/SILOptimizer/Utils/LoopUtils.cpp
@@ -33,7 +33,7 @@ static SILBasicBlock *createInitialPreheader(SILBasicBlock *Header) {
   llvm::SmallVector<SILValue, 8> Args;
   for (auto *HeaderArg : Header->getArguments()) {
     Args.push_back(Preheader->createPhiArgument(HeaderArg->getType(),
-                                                OwnershipKind::Owned));
+                                                HeaderArg->getOwnershipKind()));
   }
 
   // Create the branch to the header.

Original file line number	Diff line number	Diff line change
`@@ -33,7 +33,7 @@ static SILBasicBlock createInitialPreheader(SILBasicBlock Header) {`
`33`	`33`	`llvm::SmallVector<SILValue, 8> Args;`
`34`	`34`	`for (auto *HeaderArg : Header->getArguments()) {`
`35`	`35`	`Args.push_back(Preheader->createPhiArgument(HeaderArg->getType(),`
`36`		`- OwnershipKind::Owned));`
	`36`	`+ HeaderArg->getOwnershipKind()));`
`37`	`37`	`}`
`38`	`38`
`39`	`39`	`// Create the branch to the header.`