forked from OSchip/llvm-project
106 lines
3.9 KiB
C++
106 lines
3.9 KiB
C++
//===----- CGOpenCLRuntime.cpp - Interface to OpenCL Runtimes -------------===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This provides an abstract class for OpenCL code generation. Concrete
|
|
// subclasses of this implement code generation for specific OpenCL
|
|
// runtime libraries.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "CGOpenCLRuntime.h"
|
|
#include "CodeGenFunction.h"
|
|
#include "TargetInfo.h"
|
|
#include "llvm/IR/DerivedTypes.h"
|
|
#include "llvm/IR/GlobalValue.h"
|
|
#include <assert.h>
|
|
|
|
using namespace clang;
|
|
using namespace CodeGen;
|
|
|
|
CGOpenCLRuntime::~CGOpenCLRuntime() {}
|
|
|
|
void CGOpenCLRuntime::EmitWorkGroupLocalVarDecl(CodeGenFunction &CGF,
|
|
const VarDecl &D) {
|
|
return CGF.EmitStaticVarDecl(D, llvm::GlobalValue::InternalLinkage);
|
|
}
|
|
|
|
llvm::Type *CGOpenCLRuntime::convertOpenCLSpecificType(const Type *T) {
|
|
assert(T->isOpenCLSpecificType() &&
|
|
"Not an OpenCL specific type!");
|
|
|
|
llvm::LLVMContext& Ctx = CGM.getLLVMContext();
|
|
uint32_t ImgAddrSpc = CGM.getContext().getTargetAddressSpace(
|
|
CGM.getTarget().getOpenCLImageAddrSpace());
|
|
switch (cast<BuiltinType>(T)->getKind()) {
|
|
default:
|
|
llvm_unreachable("Unexpected opencl builtin type!");
|
|
return nullptr;
|
|
#define IMAGE_TYPE(ImgType, Id, SingletonId, Access, Suffix) \
|
|
case BuiltinType::Id: \
|
|
return llvm::PointerType::get( \
|
|
llvm::StructType::create(Ctx, "opencl." #ImgType "_" #Suffix "_t"), \
|
|
ImgAddrSpc);
|
|
#include "clang/Basic/OpenCLImageTypes.def"
|
|
case BuiltinType::OCLSampler:
|
|
return getSamplerType();
|
|
case BuiltinType::OCLEvent:
|
|
return llvm::PointerType::get(llvm::StructType::create(
|
|
Ctx, "opencl.event_t"), 0);
|
|
case BuiltinType::OCLClkEvent:
|
|
return llvm::PointerType::get(
|
|
llvm::StructType::create(Ctx, "opencl.clk_event_t"), 0);
|
|
case BuiltinType::OCLQueue:
|
|
return llvm::PointerType::get(
|
|
llvm::StructType::create(Ctx, "opencl.queue_t"), 0);
|
|
case BuiltinType::OCLReserveID:
|
|
return llvm::PointerType::get(
|
|
llvm::StructType::create(Ctx, "opencl.reserve_id_t"), 0);
|
|
}
|
|
}
|
|
|
|
llvm::Type *CGOpenCLRuntime::getPipeType() {
|
|
if (!PipeTy){
|
|
uint32_t PipeAddrSpc =
|
|
CGM.getContext().getTargetAddressSpace(LangAS::opencl_global);
|
|
PipeTy = llvm::PointerType::get(llvm::StructType::create(
|
|
CGM.getLLVMContext(), "opencl.pipe_t"), PipeAddrSpc);
|
|
}
|
|
|
|
return PipeTy;
|
|
}
|
|
|
|
llvm::PointerType *CGOpenCLRuntime::getSamplerType() {
|
|
if (!SamplerTy)
|
|
SamplerTy = llvm::PointerType::get(llvm::StructType::create(
|
|
CGM.getLLVMContext(), "opencl.sampler_t"),
|
|
CGM.getContext().getTargetAddressSpace(
|
|
LangAS::opencl_constant));
|
|
return SamplerTy;
|
|
}
|
|
|
|
llvm::Value *CGOpenCLRuntime::getPipeElemSize(const Expr *PipeArg) {
|
|
const PipeType *PipeTy = PipeArg->getType()->getAs<PipeType>();
|
|
// The type of the last (implicit) argument to be passed.
|
|
llvm::Type *Int32Ty = llvm::IntegerType::getInt32Ty(CGM.getLLVMContext());
|
|
unsigned TypeSize = CGM.getContext()
|
|
.getTypeSizeInChars(PipeTy->getElementType())
|
|
.getQuantity();
|
|
return llvm::ConstantInt::get(Int32Ty, TypeSize, false);
|
|
}
|
|
|
|
llvm::Value *CGOpenCLRuntime::getPipeElemAlign(const Expr *PipeArg) {
|
|
const PipeType *PipeTy = PipeArg->getType()->getAs<PipeType>();
|
|
// The type of the last (implicit) argument to be passed.
|
|
llvm::Type *Int32Ty = llvm::IntegerType::getInt32Ty(CGM.getLLVMContext());
|
|
unsigned TypeSize = CGM.getContext()
|
|
.getTypeAlignInChars(PipeTy->getElementType())
|
|
.getQuantity();
|
|
return llvm::ConstantInt::get(Int32Ty, TypeSize, false);
|
|
}
|