| //===----- CGOpenCLRuntime.h - Interface to OpenCL Runtimes -----*- C++ -*-===// |
| // |
| // 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. |
| // |
| //===----------------------------------------------------------------------===// |
| |
| #ifndef LLVM_CLANG_LIB_CODEGEN_CGOPENCLRUNTIME_H |
| #define LLVM_CLANG_LIB_CODEGEN_CGOPENCLRUNTIME_H |
| |
| #include "clang/AST/Type.h" |
| #include "llvm/ADT/DenseMap.h" |
| #include "llvm/IR/Type.h" |
| #include "llvm/IR/Value.h" |
| |
| namespace clang { |
| |
| class Expr; |
| class VarDecl; |
| |
| namespace CodeGen { |
| |
| class CodeGenFunction; |
| class CodeGenModule; |
| |
| class CGOpenCLRuntime { |
| protected: |
| CodeGenModule &CGM; |
| llvm::Type *PipeTy; |
| llvm::PointerType *SamplerTy; |
| |
| /// Structure for enqueued block information. |
| struct EnqueuedBlockInfo { |
| llvm::Function *Kernel; /// Enqueued block kernel. |
| llvm::Value *BlockArg; /// The first argument to enqueued block kernel. |
| }; |
| /// Maps block expression to block information. |
| llvm::DenseMap<const Expr *, EnqueuedBlockInfo> EnqueuedBlockMap; |
| |
| public: |
| CGOpenCLRuntime(CodeGenModule &CGM) : CGM(CGM), PipeTy(nullptr), |
| SamplerTy(nullptr) {} |
| virtual ~CGOpenCLRuntime(); |
| |
| /// Emit the IR required for a work-group-local variable declaration, and add |
| /// an entry to CGF's LocalDeclMap for D. The base class does this using |
| /// CodeGenFunction::EmitStaticVarDecl to emit an internal global for D. |
| virtual void EmitWorkGroupLocalVarDecl(CodeGenFunction &CGF, |
| const VarDecl &D); |
| |
| virtual llvm::Type *convertOpenCLSpecificType(const Type *T); |
| |
| virtual llvm::Type *getPipeType(const PipeType *T); |
| |
| llvm::PointerType *getSamplerType(const Type *T); |
| |
| // \brief Returnes a value which indicates the size in bytes of the pipe |
| // element. |
| virtual llvm::Value *getPipeElemSize(const Expr *PipeArg); |
| |
| // \brief Returnes a value which indicates the alignment in bytes of the pipe |
| // element. |
| virtual llvm::Value *getPipeElemAlign(const Expr *PipeArg); |
| |
| /// \return __generic void* type. |
| llvm::PointerType *getGenericVoidPointerType(); |
| |
| /// \return enqueued block information for enqueued block. |
| EnqueuedBlockInfo emitOpenCLEnqueuedBlock(CodeGenFunction &CGF, |
| const Expr *E); |
| }; |
| |
| } |
| } |
| |
| #endif |