diff options
| author | Nick Sarnie <nick.sarnie@intel.com> | 2025-06-05 10:15:38 -0400 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2025-06-05 14:15:38 +0000 |
| commit | 3b9ebe92011b033523217a9b9a2f03f4c8c37aab (patch) | |
| tree | bb519b3f1df97024db7a14874c82ae0fd65b6d7d /clang/lib/CodeGen/CodeGenModule.cpp | |
| parent | b2379bd5d59993c0d859ad90f9f5cdfcfce26e71 (diff) | |
| download | llvm-3b9ebe92011b033523217a9b9a2f03f4c8c37aab.zip llvm-3b9ebe92011b033523217a9b9a2f03f4c8c37aab.tar.gz llvm-3b9ebe92011b033523217a9b9a2f03f4c8c37aab.tar.bz2 | |
[clang] Simplify device kernel attributes (#137882)
We have multiple different attributes in clang representing device
kernels for specific targets/languages. Refactor them into one attribute
with different spellings to make it more easily scalable for new
languages/targets.
---------
Signed-off-by: Sarnie, Nick <nick.sarnie@intel.com>
Diffstat (limited to 'clang/lib/CodeGen/CodeGenModule.cpp')
| -rw-r--r-- | clang/lib/CodeGen/CodeGenModule.cpp | 11 |
1 files changed, 7 insertions, 4 deletions
diff --git a/clang/lib/CodeGen/CodeGenModule.cpp b/clang/lib/CodeGen/CodeGenModule.cpp index 468fc6e..84166dd 100644 --- a/clang/lib/CodeGen/CodeGenModule.cpp +++ b/clang/lib/CodeGen/CodeGenModule.cpp @@ -1913,7 +1913,9 @@ static std::string getMangledNameImpl(CodeGenModule &CGM, GlobalDecl GD, } else if (FD && FD->hasAttr<CUDAGlobalAttr>() && GD.getKernelReferenceKind() == KernelReferenceKind::Stub) { Out << "__device_stub__" << II->getName(); - } else if (FD && FD->hasAttr<OpenCLKernelAttr>() && + } else if (FD && + DeviceKernelAttr::isOpenCLSpelling( + FD->getAttr<DeviceKernelAttr>()) && GD.getKernelReferenceKind() == KernelReferenceKind::Stub) { Out << "__clang_ocl_kern_imp_" << II->getName(); } else { @@ -3930,7 +3932,8 @@ void CodeGenModule::EmitGlobal(GlobalDecl GD) { // Ignore declarations, they will be emitted on their first use. if (const auto *FD = dyn_cast<FunctionDecl>(Global)) { - if (FD->hasAttr<OpenCLKernelAttr>() && FD->doesThisDeclarationHaveABody()) + if (DeviceKernelAttr::isOpenCLSpelling(FD->getAttr<DeviceKernelAttr>()) && + FD->doesThisDeclarationHaveABody()) addDeferredDeclToEmit(GlobalDecl(FD, KernelReferenceKind::Stub)); // Update deferred annotations with the latest declaration if the function @@ -4895,7 +4898,7 @@ CodeGenModule::GetAddrOfFunction(GlobalDecl GD, llvm::Type *Ty, bool ForVTable, if (!Ty) { const auto *FD = cast<FunctionDecl>(GD.getDecl()); Ty = getTypes().ConvertType(FD->getType()); - if (FD->hasAttr<OpenCLKernelAttr>() && + if (DeviceKernelAttr::isOpenCLSpelling(FD->getAttr<DeviceKernelAttr>()) && GD.getKernelReferenceKind() == KernelReferenceKind::Stub) { const CGFunctionInfo &FI = getTypes().arrangeGlobalDeclaration(GD); Ty = getTypes().GetFunctionType(FI); @@ -6195,7 +6198,7 @@ void CodeGenModule::EmitGlobalFunctionDefinition(GlobalDecl GD, (CodeGenOpts.OptimizationLevel == 0) && !D->hasAttr<MinSizeAttr>(); - if (D->hasAttr<OpenCLKernelAttr>()) { + if (DeviceKernelAttr::isOpenCLSpelling(D->getAttr<DeviceKernelAttr>())) { if (GD.getKernelReferenceKind() == KernelReferenceKind::Stub && !D->hasAttr<NoInlineAttr>() && !Fn->hasFnAttribute(llvm::Attribute::NoInline) && |