contrib/llvm/tools/clang/lib/CodeGen/CGCUDARuntime.cpp

   1 //===----- CGCUDARuntime.cpp - Interface to CUDA Runtimes -----------------===//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file is distributed under the University of Illinois Open Source
   6 // License. See LICENSE.TXT for details.
   7 //
   8 //===----------------------------------------------------------------------===//
   9 //
  10 // This provides an abstract class for CUDA code generation.  Concrete
  11 // subclasses of this implement code generation for specific CUDA
  12 // runtime libraries.
  13 //
  14 //===----------------------------------------------------------------------===//
  15
  16 #include "CGCUDARuntime.h"
  17 #include "CGCall.h"
  18 #include "CodeGenFunction.h"
  19 #include "clang/AST/Decl.h"
  20 #include "clang/AST/ExprCXX.h"
  21
  22 using namespace clang;
  23 using namespace CodeGen;
  24
  25 CGCUDARuntime::~CGCUDARuntime() {}
  26
  27 RValue CGCUDARuntime::EmitCUDAKernelCallExpr(CodeGenFunction &CGF,
  28                                              const CUDAKernelCallExpr *E,
  29                                              ReturnValueSlot ReturnValue) {
  30   llvm::BasicBlock *ConfigOKBlock = CGF.createBasicBlock("kcall.configok");
  31   llvm::BasicBlock *ContBlock = CGF.createBasicBlock("kcall.end");
  32
  33   CodeGenFunction::ConditionalEvaluation eval(CGF);
  34   CGF.EmitBranchOnBoolExpr(E->getConfig(), ContBlock, ConfigOKBlock,
  35                            /*TrueCount=*/0);
  36
  37   eval.begin(CGF);
  38   CGF.EmitBlock(ConfigOKBlock);
  39
  40   const Decl *TargetDecl = nullptr;
  41   if (const ImplicitCastExpr *CE = dyn_cast<ImplicitCastExpr>(E->getCallee())) {
  42     if (const DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(CE->getSubExpr())) {
  43       TargetDecl = DRE->getDecl();
  44     }
  45   }
  46
  47   llvm::Value *Callee = CGF.EmitScalarExpr(E->getCallee());
  48   CGF.EmitCall(E->getCallee()->getType(), Callee, E->getLocStart(),
  49                ReturnValue, E->arg_begin(), E->arg_end(), TargetDecl);
  50   CGF.EmitBranch(ContBlock);
  51
  52   CGF.EmitBlock(ContBlock);
  53   eval.end(CGF);
  54
  55   return RValue::get(nullptr);
  56 }