@@ -731,10 +731,10 @@ TEST(VPBasicBlockTest, print) {
731
731
}
732
732
733
733
LLVMContext C;
734
- auto *ScalarHeader = BasicBlock::Create (C, " " );
735
- // FIXME: This looks wrong.
736
- auto ScalarHeaderVPBB = std::make_unique<VPIRBasicBlock>(ScalarHeader );
737
- VPlan Plan (VPBB0, TC, VPBB1, ScalarHeaderVPBB. get () );
734
+ auto *ScalarHeader = BasicBlock::Create (C, " scalar.header " );
735
+ auto * ScalarHeaderVPBB = new VPIRBasicBlock (ScalarHeader);
736
+ VPBlockUtils::connectBlocks (VPBB2, ScalarHeaderVPBB );
737
+ VPlan Plan (VPBB0, TC, VPBB1, ScalarHeaderVPBB);
738
738
std::string FullDump;
739
739
raw_string_ostream OS (FullDump);
740
740
Plan.printDOT (OS);
@@ -761,6 +761,11 @@ compound=true
761
761
"bb2:\l" +
762
762
" EMIT vp\<%5\> = mul vp\<%3\>, vp\<%2\>\l" +
763
763
" EMIT ret vp\<%5\>\l" +
764
+ "Successor(s): ir-bb\<scalar.header\>\l"
765
+ ]
766
+ N2 -> N3 [ label=""]
767
+ N3 [label =
768
+ "ir-bb\<scalar.header\>:\l" +
764
769
"No successors\l"
765
770
]
766
771
}
@@ -782,7 +787,7 @@ Successor(s): bb2
782
787
const char *ExpectedBlock2Str = R"( bb2:
783
788
EMIT vp<%5> = mul vp<%3>, vp<%2>
784
789
EMIT ret vp<%5>
785
- No successors
790
+ Successor(s): ir-bb<scalar.header>
786
791
)" ;
787
792
std::string Block2Dump;
788
793
raw_string_ostream OS2 (Block2Dump);
0 commit comments