@@ -533,7 +533,7 @@ namespace {
533
533
struct PartitionOpEvaluatorWithFailureCallback final
534
534
: PartitionOpEvaluatorBaseImpl<PartitionOpEvaluatorWithFailureCallback> {
535
535
using FailureCallbackTy =
536
- std::function<void (const PartitionOp &, unsigned , TransferringOperand)>;
536
+ std::function<void (const PartitionOp &, unsigned , TransferringOperand * )>;
537
537
FailureCallbackTy failureCallback;
538
538
539
539
PartitionOpEvaluatorWithFailureCallback (
@@ -543,7 +543,7 @@ struct PartitionOpEvaluatorWithFailureCallback final
543
543
failureCallback (failureCallback) {}
544
544
545
545
void handleLocalUseAfterTransfer (const PartitionOp &op, Element elt,
546
- TransferringOperand transferringOp) const {
546
+ TransferringOperand * transferringOp) const {
547
547
failureCallback (op, elt, transferringOp);
548
548
}
549
549
};
@@ -591,13 +591,13 @@ TEST(PartitionUtilsTest, TestConsumeAndRequire) {
591
591
592
592
// expected: p: ({0 1 2 6 7 10} (3 4 5) (8 9) (Element(11)))
593
593
594
- auto never_called = [](const PartitionOp &, unsigned , TransferringOperand) {
594
+ auto never_called = [](const PartitionOp &, unsigned , TransferringOperand * ) {
595
595
EXPECT_TRUE (false );
596
596
};
597
597
598
598
int times_called = 0 ;
599
599
auto increment_times_called = [&](const PartitionOp &, unsigned ,
600
- TransferringOperand) { times_called++; };
600
+ TransferringOperand * ) { times_called++; };
601
601
602
602
{
603
603
PartitionOpEvaluatorWithFailureCallback eval (p, factory,
@@ -666,7 +666,7 @@ TEST(PartitionUtilsTest, TestCopyConstructor) {
666
666
{
667
667
bool failure = false ;
668
668
PartitionOpEvaluatorWithFailureCallback eval (
669
- p1, factory, [&](const PartitionOp &, unsigned , TransferringOperand) {
669
+ p1, factory, [&](const PartitionOp &, unsigned , TransferringOperand * ) {
670
670
failure = true ;
671
671
});
672
672
eval.apply (PartitionOp::Require (Element (0 )));
@@ -675,7 +675,7 @@ TEST(PartitionUtilsTest, TestCopyConstructor) {
675
675
676
676
{
677
677
PartitionOpEvaluatorWithFailureCallback eval (
678
- p2, factory, [](const PartitionOp &, unsigned , TransferringOperand) {
678
+ p2, factory, [](const PartitionOp &, unsigned , TransferringOperand * ) {
679
679
EXPECT_TRUE (false );
680
680
});
681
681
eval.apply (PartitionOp::Require (Element (0 )));
@@ -688,7 +688,7 @@ TEST(PartitionUtilsTest, TestUndoTransfer) {
688
688
689
689
Partition p;
690
690
PartitionOpEvaluatorWithFailureCallback eval (
691
- p, factory, [&](const PartitionOp &, unsigned , TransferringOperand) {
691
+ p, factory, [&](const PartitionOp &, unsigned , TransferringOperand * ) {
692
692
EXPECT_TRUE (false );
693
693
});
694
694
0 commit comments