offload/include/OffloadEntry.h

   1 //===-- OffloadEntry.h - Representation of offload entries ------*- C++ -*-===//
   2 //
   3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
   4 // See https://llvm.org/LICENSE.txt for license information.
   5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
   6 //
   7 //===----------------------------------------------------------------------===//
   8 //
   9 //
  10 //===----------------------------------------------------------------------===//
  11
  12 #ifndef OMPTARGET_OFFLOAD_ENTRY_H
  13 #define OMPTARGET_OFFLOAD_ENTRY_H
  14
  15 #include "Shared/APITypes.h"
  16
  17 #include "omptarget.h"
  18
  19 #include "llvm/ADT/StringRef.h"
  20
  21 class DeviceImageTy;
  22
  23 class OffloadEntryTy {
  24   DeviceImageTy &DeviceImage;
  25   __tgt_offload_entry &OffloadEntry;
  26
  27 public:
  28   OffloadEntryTy(DeviceImageTy &DeviceImage, __tgt_offload_entry &OffloadEntry)
  29       : DeviceImage(DeviceImage), OffloadEntry(OffloadEntry) {}
  30
  31   bool isGlobal() const { return getSize() != 0; }
  32   size_t getSize() const { return OffloadEntry.size; }
  33
  34   void *getAddress() const { return OffloadEntry.addr; }
  35   llvm::StringRef getName() const { return OffloadEntry.name; }
  36   const char *getNameAsCStr() const { return OffloadEntry.name; }
  37   __tgt_bin_desc *getBinaryDescription() const;
  38
  39   bool isLink() const { return hasFlags(OMP_DECLARE_TARGET_LINK); }
  40
  41   bool hasFlags(OpenMPOffloadingDeclareTargetFlags Flags) const {
  42     return Flags & OffloadEntry.flags;
  43   }
  44 };
  45
  46 #endif // OMPTARGET_OFFLOAD_ENTRY_H