openmp/libomptarget/plugins/remote/src/rtl.cpp

   1 //===--------------------- rtl.cpp - Remote RTL Plugin --------------------===//
   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 // RTL for Host.
  10 //
  11 //===----------------------------------------------------------------------===//
  12
  13 #include <cstddef>
  14 #include <memory>
  15 #include <string>
  16
  17 #include "Client.h"
  18 #include "Utils.h"
  19 #include "omptarget.h"
  20 #include "omptargetplugin.h"
  21
  22 #define TARGET_NAME RPC
  23 #define DEBUG_PREFIX "Target " GETNAME(TARGET_NAME) " RTL"
  24
  25 RemoteClientManager *Manager;
  26
  27 __attribute__((constructor(101))) void initRPC() {
  28   DP("Init RPC library!\n");
  29
  30   Manager = new RemoteClientManager();
  31 }
  32
  33 __attribute__((destructor(101))) void deinitRPC() {
  34   Manager->shutdown(); // TODO: Error handle shutting down
  35   DP("Deinit RPC library!\n");
  36   delete Manager;
  37 }
  38
  39 // Exposed library API function
  40 #ifdef __cplusplus
  41 extern "C" {
  42 #endif
  43
  44 int32_t __tgt_rtl_register_lib(__tgt_bin_desc *Desc) {
  45   return Manager->registerLib(Desc);
  46 }
  47
  48 int32_t __tgt_rtl_unregister_lib(__tgt_bin_desc *Desc) {
  49   return Manager->unregisterLib(Desc);
  50 }
  51
  52 int32_t __tgt_rtl_is_valid_binary(__tgt_device_image *Image) {
  53   return Manager->isValidBinary(Image);
  54 }
  55
  56 int32_t __tgt_rtl_number_of_devices() { return Manager->getNumberOfDevices(); }
  57
  58 int32_t __tgt_rtl_init_device(int32_t DeviceId) {
  59   return Manager->initDevice(DeviceId);
  60 }
  61
  62 int64_t __tgt_rtl_init_requires(int64_t RequiresFlags) {
  63   return Manager->initRequires(RequiresFlags);
  64 }
  65
  66 __tgt_target_table *__tgt_rtl_load_binary(int32_t DeviceId,
  67                                           __tgt_device_image *Image) {
  68   return Manager->loadBinary(DeviceId, (__tgt_device_image *)Image);
  69 }
  70
  71 int32_t __tgt_rtl_is_data_exchangable(int32_t SrcDevId, int32_t DstDevId) {
  72   return Manager->isDataExchangeable(SrcDevId, DstDevId);
  73 }
  74
  75 void *__tgt_rtl_data_alloc(int32_t DeviceId, int64_t Size, void *HstPtr,
  76                            int32_t Kind) {
  77   if (Kind != TARGET_ALLOC_DEFAULT) {
  78     REPORT("Invalid target data allocation kind or requested allocator not "
  79            "implemented yet\n");
  80     return NULL;
  81   }
  82
  83   return Manager->dataAlloc(DeviceId, Size, HstPtr);
  84 }
  85
  86 int32_t __tgt_rtl_data_submit(int32_t DeviceId, void *TgtPtr, void *HstPtr,
  87                               int64_t Size) {
  88   return Manager->dataSubmit(DeviceId, TgtPtr, HstPtr, Size);
  89 }
  90
  91 int32_t __tgt_rtl_data_retrieve(int32_t DeviceId, void *HstPtr, void *TgtPtr,
  92                                 int64_t Size) {
  93   return Manager->dataRetrieve(DeviceId, HstPtr, TgtPtr, Size);
  94 }
  95
  96 int32_t __tgt_rtl_data_delete(int32_t DeviceId, void *TgtPtr) {
  97   return Manager->dataDelete(DeviceId, TgtPtr);
  98 }
  99
 100 int32_t __tgt_rtl_data_exchange(int32_t SrcDevId, void *SrcPtr,
 101                                 int32_t DstDevId, void *DstPtr, int64_t Size) {
 102   return Manager->dataExchange(SrcDevId, SrcPtr, DstDevId, DstPtr, Size);
 103 }
 104
 105
 106 int32_t __tgt_rtl_run_target_region(int32_t DeviceId, void *TgtEntryPtr,
 107                                     void **TgtArgs, ptrdiff_t *TgtOffsets,
 108                                     int32_t ArgNum) {
 109   return Manager->runTargetRegion(DeviceId, TgtEntryPtr, TgtArgs, TgtOffsets,
 110                                   ArgNum);
 111 }
 112
 113 int32_t __tgt_rtl_run_target_team_region(int32_t DeviceId, void *TgtEntryPtr,
 114                                          void **TgtArgs, ptrdiff_t *TgtOffsets,
 115                                          int32_t ArgNum, int32_t TeamNum,
 116                                          int32_t ThreadLimit,
 117                                          uint64_t LoopTripCount) {
 118   return Manager->runTargetTeamRegion(DeviceId, TgtEntryPtr, TgtArgs,
 119                                       TgtOffsets, ArgNum, TeamNum, ThreadLimit,
 120                                       LoopTripCount);
 121 }
 122
 123 // Exposed library API function
 124 #ifdef __cplusplus
 125 }
 126 #endif