CIRGenOpenMPRuntime.h

Go to the documentation of this file.

//===--- CIRGenOpenMPRuntime.h - OpenMP code generation helpers -----------===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
 
#ifndef LLVM_CLANG_LIB_CIR_CODEGEN_CIRGENOPENMPRUNTIME_H
#define LLVM_CLANG_LIB_CIR_CODEGEN_CIRGENOPENMPRUNTIME_H
 
#include "clang/AST/DeclOpenMP.h"
#include "clang/AST/GlobalDecl.h"
#include "clang/AST/StmtOpenMP.h"
#include "clang/CIR/Dialect/IR/CIRDialect.h"
#include "llvm/ADT/DenseSet.h"
 
namespace clang::CIRGen {
 
class CIRGenModule;
 
class CIRGenOpenMPRuntime {
  CIRGenModule &cgm;
 
  /// Declarations that have been force-emitted for the target device because
  /// they are transitively referenced from declare target functions.
  llvm::DenseSet<CanonicalDeclPtr<const Decl>> alreadyEmittedTargetDecls;
 
  /// Returns false if the given function or declare reduction should be
  /// emitted. Returns true if it should eb skipped.
  /// emission).
  bool emitTargetFunctions(GlobalDecl gd);
 
  /// Returns false if given global variable should be emitted. Returns
  /// true if it should be skipped.
  bool emitTargetGlobalVariable(GlobalDecl gd);
 
public:
  explicit CIRGenOpenMPRuntime(CIRGenModule &cgm) : cgm(cgm) {}
 
  /// Check whether the given GlobalDecl needs special handling for device
  /// compilation. Returns false if it should be emitted, true if it should be
  /// skipped.
  bool emitTargetGlobal(GlobalDecl gd);
 
  /// Mark a function reference as one that should be emitted on the device.
  /// Returns false if it should be emitted, true if the function is already
  /// handled and should be skipped.
  bool markAsGlobalTarget(GlobalDecl gd);
 
  /// If the function has an OMPDeclareTargetDeclAttr, set the corresponding
  /// omp.declare_target attribute on the emitted cir.func op.
  void emitDeclareTargetFunction(const FunctionDecl *fd, cir::FuncOp funcOp);
};
 
} // namespace clang::CIRGen
 
#endif // LLVM_CLANG_LIB_CIR_CODEGEN_CIRGENOPENMPRUNTIME_H