44 lines
1.2 KiB
C++
44 lines
1.2 KiB
C++
//===-- 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
|