Skip to content

Commit 9dd341e

Browse files
committed
fix test failure
1 parent 9219eba commit 9dd341e

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

llvm/tools/sycl-post-link/sycl-post-link.cpp

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -562,7 +562,7 @@ module_split::ModuleDesc link(module_split::ModuleDesc &&MD1,
562562
}
563563
module_split::ModuleDesc Res(MD1.releaseModulePtr(), std::move(Names));
564564
Res.assignMergedProperties(MD1, MD2);
565-
Res.Name = "linked[" + MD1.Name + "," + MD1.Name + "]";
565+
Res.Name = "linked[" + MD1.Name + "," + MD2.Name + "]";
566566
return Res;
567567
}
568568

@@ -789,9 +789,13 @@ processInputModule(std::unique_ptr<Module> M) {
789789
module_split::ModuleDesc M2 =
790790
link(std::move(MMs[0]), std::move(MMs[1]));
791791
M2.restoreLinkageOfDirectInvokeSimdTargets();
792-
M2.cleanup();
792+
string_vector Names;
793+
M2.saveEntryPointNames(Names);
794+
M2.cleanup(); // may remove some entry points, need to save/rebuild
795+
M2.rebuildEntryPoints(Names);
793796
MMs.clear();
794797
MMs.emplace_back(std::move(M2));
798+
DUMP_ENTRY_POINTS(MMs.back().entries(), MMs.back().Name.c_str(), 3);
795799
Modified = true;
796800
}
797801
bool SplitOccurred = SplitByScope || SplitByDoubleGRF || SplitByESIMD;

0 commit comments

Comments
 (0)