Skip to content

Commit aafd9f4

Browse files
committed
Fix unittest
1 parent a5bd7a2 commit aafd9f4

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

unittests/SILOptimizer/PartitionUtilsTest.cpp

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -533,7 +533,7 @@ namespace {
533533
struct PartitionOpEvaluatorWithFailureCallback final
534534
: PartitionOpEvaluatorBaseImpl<PartitionOpEvaluatorWithFailureCallback> {
535535
using FailureCallbackTy =
536-
std::function<void(const PartitionOp &, unsigned, TransferringOperand)>;
536+
std::function<void(const PartitionOp &, unsigned, TransferringOperand *)>;
537537
FailureCallbackTy failureCallback;
538538

539539
PartitionOpEvaluatorWithFailureCallback(
@@ -543,7 +543,7 @@ struct PartitionOpEvaluatorWithFailureCallback final
543543
failureCallback(failureCallback) {}
544544

545545
void handleLocalUseAfterTransfer(const PartitionOp &op, Element elt,
546-
TransferringOperand transferringOp) const {
546+
TransferringOperand *transferringOp) const {
547547
failureCallback(op, elt, transferringOp);
548548
}
549549
};
@@ -591,13 +591,13 @@ TEST(PartitionUtilsTest, TestConsumeAndRequire) {
591591

592592
// expected: p: ({0 1 2 6 7 10} (3 4 5) (8 9) (Element(11)))
593593

594-
auto never_called = [](const PartitionOp &, unsigned, TransferringOperand) {
594+
auto never_called = [](const PartitionOp &, unsigned, TransferringOperand *) {
595595
EXPECT_TRUE(false);
596596
};
597597

598598
int times_called = 0;
599599
auto increment_times_called = [&](const PartitionOp &, unsigned,
600-
TransferringOperand) { times_called++; };
600+
TransferringOperand *) { times_called++; };
601601

602602
{
603603
PartitionOpEvaluatorWithFailureCallback eval(p, factory,
@@ -666,7 +666,7 @@ TEST(PartitionUtilsTest, TestCopyConstructor) {
666666
{
667667
bool failure = false;
668668
PartitionOpEvaluatorWithFailureCallback eval(
669-
p1, factory, [&](const PartitionOp &, unsigned, TransferringOperand) {
669+
p1, factory, [&](const PartitionOp &, unsigned, TransferringOperand *) {
670670
failure = true;
671671
});
672672
eval.apply(PartitionOp::Require(Element(0)));
@@ -675,7 +675,7 @@ TEST(PartitionUtilsTest, TestCopyConstructor) {
675675

676676
{
677677
PartitionOpEvaluatorWithFailureCallback eval(
678-
p2, factory, [](const PartitionOp &, unsigned, TransferringOperand) {
678+
p2, factory, [](const PartitionOp &, unsigned, TransferringOperand *) {
679679
EXPECT_TRUE(false);
680680
});
681681
eval.apply(PartitionOp::Require(Element(0)));
@@ -688,7 +688,7 @@ TEST(PartitionUtilsTest, TestUndoTransfer) {
688688

689689
Partition p;
690690
PartitionOpEvaluatorWithFailureCallback eval(
691-
p, factory, [&](const PartitionOp &, unsigned, TransferringOperand) {
691+
p, factory, [&](const PartitionOp &, unsigned, TransferringOperand *) {
692692
EXPECT_TRUE(false);
693693
});
694694

0 commit comments

Comments
 (0)