intel/llvm
1
0
Fork 0
mirror of https://github.com/intel/llvm.git synced 2026-01-18 07:57:36 +08:00
Code Issues Packages Projects Releases Wiki Activity

[clang][CodeGenOpenCL][NFC] Remove redundant map lookups (#125285)

Browse Source
This commit is contained in:
Balazs Benics
2025-02-01 08:21:15 +01:00
committed by GitHub
parent 0d21ef4e6c
commit 65708bad57
1 changed files with 13 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
clang/lib/CodeGen/CGOpenCLRuntime.cpp
View File

@@ -130,10 +130,11 @@ void CGOpenCLRuntime::recordBlockInfo(const BlockExpr *E,
assert(!EnqueuedBlockMap.contains(E) && "Block expression emitted twice");
assert(isa<llvm::Function>(InvokeF) && "Invalid invoke function");
assert(Block->getType()->isPointerTy() && "Invalid block literal type");
EnqueuedBlockMap[E].InvokeFunc = InvokeF;
EnqueuedBlockMap[E].BlockArg = Block;
EnqueuedBlockMap[E].BlockTy = BlockTy;
EnqueuedBlockMap[E].KernelHandle = nullptr;
EnqueuedBlockInfo &BlockInfo = EnqueuedBlockMap[E];
BlockInfo.InvokeFunc = InvokeF;
BlockInfo.BlockArg = Block;
BlockInfo.BlockTy = BlockTy;
BlockInfo.KernelHandle = nullptr;
}
llvm::Function *CGOpenCLRuntime::getInvokeFunction(const Expr *E) {
@@ -148,17 +149,19 @@ CGOpenCLRuntime::emitOpenCLEnqueuedBlock(CodeGenFunction &CGF, const Expr *E) {
// to get the block literal.
const BlockExpr *Block = getBlockExpr(E);
assert(EnqueuedBlockMap.contains(Block) && "Block expression not emitted");
auto It = EnqueuedBlockMap.find(Block);
assert(It != EnqueuedBlockMap.end() && "Block expression not emitted");
EnqueuedBlockInfo &BlockInfo = It->second;
// Do not emit the block wrapper again if it has been emitted.
if (EnqueuedBlockMap[Block].KernelHandle) {
return EnqueuedBlockMap[Block];
if (BlockInfo.KernelHandle) {
return BlockInfo;
}
auto *F = CGF.getTargetHooks().createEnqueuedBlockKernel(
CGF, EnqueuedBlockMap[Block].InvokeFunc, EnqueuedBlockMap[Block].BlockTy);
CGF, BlockInfo.InvokeFunc, BlockInfo.BlockTy);
// The common part of the post-processing of the kernel goes here.
EnqueuedBlockMap[Block].KernelHandle = F;
return EnqueuedBlockMap[Block];
BlockInfo.KernelHandle = F;
return BlockInfo;
}