Home | History | Annotate | Download | only in CodeGen
      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, ReturnValue, TargetDecl);
     49   CGF.EmitBranch(ContBlock);
     50 
     51   CGF.EmitBlock(ContBlock);
     52   eval.end(CGF);
     53 
     54   return RValue::get(nullptr);
     55 }
     56