diff options
author | Alexey Bataev <a.bataev@hotmail.com> | 2018-07-09 17:46:55 +0000 |
---|---|---|
committer | Alexey Bataev <a.bataev@hotmail.com> | 2018-07-09 17:46:55 +0000 |
commit | ebe25f327bae857aeffdfb72aebc91db9ab3b218 (patch) | |
tree | 54b5fafb959cc69d9968ab06a96831fc64b0264b | |
parent | d76271da05f02da57940ffbe8b832d34c14bf6a5 (diff) |
[OPENMP, NVPTX] Support several images in the executable.
Summary:
Currently Cuda plugin supports loading of the single image, though we
may have the executable with the several images, if it has target
regions inside of the dynamically loaded library. Patch allows to load
multiple images.
Reviewers: grokos
Subscribers: guansong, openmp-commits, kkwli0
Differential Revision: https://reviews.llvm.org/D49036
git-svn-id: https://llvm.org/svn/llvm-project/openmp/trunk@336569 91177308-0d34-0410-b5e6-96231b3b80d8
-rw-r--r-- | libomptarget/plugins/cuda/src/rtl.cpp | 11 |
1 files changed, 6 insertions, 5 deletions
diff --git a/libomptarget/plugins/cuda/src/rtl.cpp b/libomptarget/plugins/cuda/src/rtl.cpp index fe2f9f6..872e7f0 100644 --- a/libomptarget/plugins/cuda/src/rtl.cpp +++ b/libomptarget/plugins/cuda/src/rtl.cpp @@ -92,7 +92,7 @@ std::list<KernelTy> KernelsList; /// Class containing all the device information. class RTLDeviceInfoTy { - std::vector<FuncOrGblEntryTy> FuncGblEntries; + std::vector<std::list<FuncOrGblEntryTy>> FuncGblEntries; public: int NumberOfDevices; @@ -122,7 +122,7 @@ public: void addOffloadEntry(int32_t device_id, __tgt_offload_entry entry) { assert(device_id < (int32_t)FuncGblEntries.size() && "Unexpected device id!"); - FuncOrGblEntryTy &E = FuncGblEntries[device_id]; + FuncOrGblEntryTy &E = FuncGblEntries[device_id].back(); E.Entries.push_back(entry); } @@ -131,7 +131,7 @@ public: bool findOffloadEntry(int32_t device_id, void *addr) { assert(device_id < (int32_t)FuncGblEntries.size() && "Unexpected device id!"); - FuncOrGblEntryTy &E = FuncGblEntries[device_id]; + FuncOrGblEntryTy &E = FuncGblEntries[device_id].back(); for (auto &it : E.Entries) { if (it.addr == addr) @@ -145,7 +145,7 @@ public: __tgt_target_table *getOffloadEntriesTable(int32_t device_id) { assert(device_id < (int32_t)FuncGblEntries.size() && "Unexpected device id!"); - FuncOrGblEntryTy &E = FuncGblEntries[device_id]; + FuncOrGblEntryTy &E = FuncGblEntries[device_id].back(); int32_t size = E.Entries.size(); @@ -167,7 +167,8 @@ public: void clearOffloadEntriesTable(int32_t device_id) { assert(device_id < (int32_t)FuncGblEntries.size() && "Unexpected device id!"); - FuncOrGblEntryTy &E = FuncGblEntries[device_id]; + FuncGblEntries[device_id].emplace_back(); + FuncOrGblEntryTy &E = FuncGblEntries[device_id].back(); E.Entries.clear(); E.Table.EntriesBegin = E.Table.EntriesEnd = 0; } |