621bafd5c1
Summary: This patch removes the bulk of the handling of the `__tgt_offload_entries` out of the plugins itself. The reason for this is because the plugins themselves should not be handling this implementation detail of the OpenMP runtime. Instead, we expose two new plugin API functions to get the points to a device pointer for a global as well as a kernel type. This required introducing a new type to represent a binary image that has been loaded on a device. We can then use this to load the addresses as needed. The creation of the mapping table is then handled just in `libomptarget` where we simply look up each address individually. This should allow us to expose these operations more generically when we provide a separate API.
57 lines
1.9 KiB
C++
57 lines
1.9 KiB
C++
//===------------ rtl.h - Target independent OpenMP target RTL ------------===//
|
|
//
|
|
// 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// Declarations for handling RTL plugins.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef _OMPTARGET_RTL_H
|
|
#define _OMPTARGET_RTL_H
|
|
|
|
#include "llvm/ADT/SmallVector.h"
|
|
|
|
#include "omptarget.h"
|
|
|
|
#include <cstdint>
|
|
#include <map>
|
|
|
|
/// Map between the host entry begin and the translation table. Each
|
|
/// registered library gets one TranslationTable. Use the map from
|
|
/// __tgt_offload_entry so that we may quickly determine whether we
|
|
/// are trying to (re)register an existing lib or really have a new one.
|
|
struct TranslationTable {
|
|
__tgt_target_table HostTable;
|
|
llvm::SmallVector<__tgt_target_table> DeviceTables;
|
|
|
|
// Image assigned to a given device.
|
|
llvm::SmallVector<__tgt_device_image *>
|
|
TargetsImages; // One image per device ID.
|
|
|
|
// Arrays of entries active on the device.
|
|
llvm::SmallVector<llvm::SmallVector<__tgt_offload_entry>>
|
|
TargetsEntries; // One table per device ID.
|
|
|
|
// Table of entry points or NULL if it was not already computed.
|
|
llvm::SmallVector<__tgt_target_table *>
|
|
TargetsTable; // One table per device ID.
|
|
};
|
|
typedef std::map<__tgt_offload_entry *, TranslationTable>
|
|
HostEntriesBeginToTransTableTy;
|
|
|
|
/// Map between the host ptr and a table index
|
|
struct TableMap {
|
|
TranslationTable *Table = nullptr; // table associated with the host ptr.
|
|
uint32_t Index = 0; // index in which the host ptr translated entry is found.
|
|
TableMap() = default;
|
|
TableMap(TranslationTable *Table, uint32_t Index)
|
|
: Table(Table), Index(Index) {}
|
|
};
|
|
typedef std::map<void *, TableMap> HostPtrToTableMapTy;
|
|
|
|
#endif
|