1//===- TCE.cpp ------------------------------------------------------------===//
2//
3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4// See https://llvm.org/LICENSE.txt for license information.
5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6//
7//===----------------------------------------------------------------------===//
8
9#include "ABIInfoImpl.h"
10#include "TargetInfo.h"
11
12using namespace clang;
13using namespace clang::CodeGen;
14
15//===----------------------------------------------------------------------===//
16// TCE ABI Implementation (see http://tce.cs.tut.fi). Uses mostly the defaults.
17// Currently subclassed only to implement custom OpenCL C function attribute
18// handling.
19//===----------------------------------------------------------------------===//
20
21namespace {
22
23class TCETargetCodeGenInfo : public TargetCodeGenInfo {
24public:
25 TCETargetCodeGenInfo(CodeGenTypes &CGT)
26 : TargetCodeGenInfo(std::make_unique<DefaultABIInfo>(args&: CGT)) {}
27
28 void setTargetAttributes(const Decl *D, llvm::GlobalValue *GV,
29 CodeGen::CodeGenModule &M) const override;
30};
31
32void TCETargetCodeGenInfo::setTargetAttributes(
33 const Decl *D, llvm::GlobalValue *GV, CodeGen::CodeGenModule &M) const {
34 if (GV->isDeclaration())
35 return;
36 const FunctionDecl *FD = dyn_cast_or_null<FunctionDecl>(Val: D);
37 if (!FD) return;
38
39 llvm::Function *F = cast<llvm::Function>(Val: GV);
40
41 if (M.getLangOpts().OpenCL) {
42 if (FD->hasAttr<OpenCLKernelAttr>()) {
43 // OpenCL C Kernel functions are not subject to inlining
44 F->addFnAttr(Kind: llvm::Attribute::NoInline);
45 const ReqdWorkGroupSizeAttr *Attr = FD->getAttr<ReqdWorkGroupSizeAttr>();
46 if (Attr) {
47 // Convert the reqd_work_group_size() attributes to metadata.
48 llvm::LLVMContext &Context = F->getContext();
49 llvm::NamedMDNode *OpenCLMetadata =
50 M.getModule().getOrInsertNamedMetadata(
51 Name: "opencl.kernel_wg_size_info");
52
53 SmallVector<llvm::Metadata *, 5> Operands;
54 Operands.push_back(Elt: llvm::ConstantAsMetadata::get(C: F));
55
56 Operands.push_back(
57 Elt: llvm::ConstantAsMetadata::get(C: llvm::Constant::getIntegerValue(
58 Ty: M.Int32Ty, V: llvm::APInt(32, Attr->getXDim()))));
59 Operands.push_back(
60 Elt: llvm::ConstantAsMetadata::get(C: llvm::Constant::getIntegerValue(
61 Ty: M.Int32Ty, V: llvm::APInt(32, Attr->getYDim()))));
62 Operands.push_back(
63 Elt: llvm::ConstantAsMetadata::get(C: llvm::Constant::getIntegerValue(
64 Ty: M.Int32Ty, V: llvm::APInt(32, Attr->getZDim()))));
65
66 // Add a boolean constant operand for "required" (true) or "hint"
67 // (false) for implementing the work_group_size_hint attr later.
68 // Currently always true as the hint is not yet implemented.
69 Operands.push_back(
70 Elt: llvm::ConstantAsMetadata::get(C: llvm::ConstantInt::getTrue(Context)));
71 OpenCLMetadata->addOperand(M: llvm::MDNode::get(Context, MDs: Operands));
72 }
73 }
74 }
75}
76
77}
78
79std::unique_ptr<TargetCodeGenInfo>
80CodeGen::createTCETargetCodeGenInfo(CodeGenModule &CGM) {
81 return std::make_unique<TCETargetCodeGenInfo>(args&: CGM.getTypes());
82}
83