@@ -533,7 +533,7 @@ namespace {
533533struct 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