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
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
|
//===-------------------------- Server.h - Server -------------------------===//
//
// 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
//
//===----------------------------------------------------------------------===//
//
// Offloading gRPC server for remote host.
//
//===----------------------------------------------------------------------===//
#ifndef LLVM_OPENMP_LIBOMPTARGET_PLUGINS_REMOTE_SERVER_SERVER_H
#define LLVM_OPENMP_LIBOMPTARGET_PLUGINS_REMOTE_SERVER_SERVER_H
#include <grpcpp/server_context.h>
#include "Utils.h"
#include "device.h"
#include "omptarget.h"
#include "openmp.grpc.pb.h"
#include "openmp.pb.h"
#include "rtl.h"
using grpc::ServerContext;
using grpc::ServerReader;
using grpc::ServerWriter;
using grpc::Status;
using namespace openmp::libomptarget::remote;
using namespace RemoteOffloading;
using namespace google;
extern PluginManager *PM;
class RemoteOffloadImpl final : public RemoteOffload::Service {
private:
int32_t mapHostRTLDeviceId(int32_t RTLDeviceID);
std::unordered_map<const void *, __tgt_device_image *>
HostToRemoteDeviceImage;
std::unordered_map<const void *, std::unique_ptr<__tgt_bin_desc>>
Descriptions;
__tgt_target_table *Table = nullptr;
int DebugLevel;
uint64_t MaxSize;
uint64_t BlockSize;
std::unique_ptr<protobuf::Arena> Arena;
public:
RemoteOffloadImpl(uint64_t MaxSize, uint64_t BlockSize)
: MaxSize(MaxSize), BlockSize(BlockSize) {
DebugLevel = getDebugLevel();
Arena = std::make_unique<protobuf::Arena>();
}
Status Shutdown(ServerContext *Context, const Null *Request,
I32 *Reply) override;
Status RegisterLib(ServerContext *Context,
const TargetBinaryDescription *Description,
I32 *Reply) override;
Status UnregisterLib(ServerContext *Context, const Pointer *Request,
I32 *Reply) override;
Status IsValidBinary(ServerContext *Context,
const TargetDeviceImagePtr *Image,
I32 *IsValid) override;
Status GetNumberOfDevices(ServerContext *Context, const Null *Null,
I32 *NumberOfDevices) override;
Status InitDevice(ServerContext *Context, const I32 *DeviceNum,
I32 *Reply) override;
Status InitRequires(ServerContext *Context, const I64 *RequiresFlag,
I32 *Reply) override;
Status LoadBinary(ServerContext *Context, const Binary *Binary,
TargetTable *Reply) override;
Status IsDataExchangeable(ServerContext *Context, const DevicePair *Request,
I32 *Reply) override;
Status DataAlloc(ServerContext *Context, const AllocData *Request,
Pointer *Reply) override;
Status DataSubmit(ServerContext *Context, ServerReader<SubmitData> *Reader,
I32 *Reply) override;
Status DataRetrieve(ServerContext *Context, const RetrieveData *Request,
ServerWriter<Data> *Writer) override;
Status DataExchange(ServerContext *Context, const ExchangeData *Request,
I32 *Reply) override;
Status DataDelete(ServerContext *Context, const DeleteData *Request,
I32 *Reply) override;
Status RunTargetRegion(ServerContext *Context, const TargetRegion *Request,
I32 *Reply) override;
Status RunTargetTeamRegion(ServerContext *Context,
const TargetTeamRegion *Request,
I32 *Reply) override;
};
#endif
|
