aboutsummaryrefslogtreecommitdiff
path: root/offload/include/DeviceImage.h
blob: ba46a837f402cf7ca673df2a6af84c97ec6c4d6c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
//===-- DeviceImage.h - Representation of the device code/image -*- C++ -*-===//
//
// 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 OMPTARGET_DEVICE_IMAGE_H
#define OMPTARGET_DEVICE_IMAGE_H

#include "OffloadEntry.h"
#include "Shared/APITypes.h"

#include "llvm/ADT/SmallVector.h"
#include "llvm/ADT/StringRef.h"
#include "llvm/ADT/iterator.h"
#include "llvm/ADT/iterator_range.h"
#include "llvm/Object/OffloadBinary.h"

#include <memory>

class DeviceImageTy {

  std::unique_ptr<llvm::object::OffloadBinary> Binary;

  __tgt_bin_desc *BinaryDesc;
  __tgt_device_image Image;

public:
  DeviceImageTy(__tgt_bin_desc &BinaryDesc, __tgt_device_image &Image);

  __tgt_device_image &getExecutableImage() { return Image; }
  __tgt_bin_desc &getBinaryDesc() { return *BinaryDesc; }

  auto entries() {
    return llvm::make_range(Image.EntriesBegin, Image.EntriesEnd);
  }
};

#endif // OMPTARGET_DEVICE_IMAGE_H