nicolas/llvm-project
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Patch by Anton Lokhmotov to add OpenCL work group size attributes.

Browse Source 
llvm-svn: 159965
This commit is contained in:
Tanya Lattner
2012-07-09 22:06:01 +00:00
parent 17709ae8d9
commit bcffcdfd18
6 changed files with 121 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

54
clang/lib/CodeGen/CodeGenFunction.cpp
View File

@@ -252,6 +252,51 @@ void CodeGenFunction::EmitMCountInstrumentation() {
Builder.CreateCall(MCountFn);
}
void CodeGenFunction::EmitOpenCLKernelMetadata(const FunctionDecl *FD,
llvm::Function *Fn)
{
if (!FD->hasAttr<OpenCLKernelAttr>())
return;
llvm::LLVMContext &Context = getLLVMContext();
llvm::SmallVector <llvm::Value*, 5> kernelMDArgs;
kernelMDArgs.push_back(Fn);
if (FD->hasAttr<WorkGroupSizeHintAttr>()) {
llvm::SmallVector <llvm::Value*, 5> attrMDArgs;
attrMDArgs.push_back(llvm::MDString::get(Context, "work_group_size_hint"));
WorkGroupSizeHintAttr *attr = FD->getAttr<WorkGroupSizeHintAttr>();
llvm::Type *iTy = llvm::IntegerType::get(Context, 32);
attrMDArgs.push_back(llvm::ConstantInt::get(iTy,
llvm::APInt(32, (uint64_t)attr->getXDim())));
attrMDArgs.push_back(llvm::ConstantInt::get(iTy,
llvm::APInt(32, (uint64_t)attr->getYDim())));
attrMDArgs.push_back(llvm::ConstantInt::get(iTy,
llvm::APInt(32, (uint64_t)attr->getZDim())));
kernelMDArgs.push_back(llvm::MDNode::get(Context, attrMDArgs));
}
if (FD->hasAttr<ReqdWorkGroupSizeAttr>()) {
llvm::SmallVector <llvm::Value*, 5> attrMDArgs;
attrMDArgs.push_back(llvm::MDString::get(Context, "reqd_work_group_size"));
ReqdWorkGroupSizeAttr *attr = FD->getAttr<ReqdWorkGroupSizeAttr>();
llvm::Type *iTy = llvm::IntegerType::get(Context, 32);
attrMDArgs.push_back(llvm::ConstantInt::get(iTy,
llvm::APInt(32, (uint64_t)attr->getXDim())));
attrMDArgs.push_back(llvm::ConstantInt::get(iTy,
llvm::APInt(32, (uint64_t)attr->getYDim())));
attrMDArgs.push_back(llvm::ConstantInt::get(iTy,
llvm::APInt(32, (uint64_t)attr->getZDim())));
kernelMDArgs.push_back(llvm::MDNode::get(Context, attrMDArgs));
}
llvm::MDNode *kernelMDNode = llvm::MDNode::get(Context, kernelMDArgs);
llvm::NamedMDNode *OpenCLKernelMetadata =
CGM.getModule().getOrInsertNamedMetadata("opencl.kernels");
OpenCLKernelMetadata->addOperand(kernelMDNode);
}
void CodeGenFunction::StartFunction(GlobalDecl GD, QualType RetTy,
llvm::Function *Fn,
const CGFunctionInfo &FnInfo,
@@ -280,14 +325,7 @@ void CodeGenFunction::StartFunction(GlobalDecl GD, QualType RetTy,
if (getContext().getLangOpts().OpenCL) {
// Add metadata for a kernel function.
if (const FunctionDecl *FD = dyn_cast_or_null<FunctionDecl>(D))
if (FD->hasAttr<OpenCLKernelAttr>()) {
llvm::LLVMContext &Context = getLLVMContext();
llvm::NamedMDNode *OpenCLMetadata =
CGM.getModule().getOrInsertNamedMetadata("opencl.kernels");
llvm::Value *Op = Fn;
OpenCLMetadata->addOperand(llvm::MDNode::get(Context, Op));
}
EmitOpenCLKernelMetadata(FD, Fn);
}
llvm::BasicBlock *EntryBB = createBasicBlock("entry", CurFn);