ec8f4a38c8
This introduces a remote offloading plugin for libomptarget. This implementation relies on gRPC and protobuf, so this library will only build if both libraries are available on the system. The corresponding server is compiled to `openmp-offloading-server`. This is a large change, but the only way to split this up is into RTL/server but I fear that could introduce an inconsistency amongst them. Ideally, tests for this should be added to the current ones that but that is problematic for at least one reason. Given that libomptarget registers plugin on a first-come-first-serve basis, if we wanted to offload onto a local x86 through a different process, then we'd have to either re-order the plugin list in `rtl.cpp` (which is what I did locally for testing) or find a better solution for runtime plugin registration in libomptarget. Differential Revision: https://reviews.llvm.org/D95314
53 lines
1.4 KiB
C++
53 lines
1.4 KiB
C++
//===------------- OffloadingServer.cpp - Server Application --------------===//
|
|
//
|
|
// 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 server for remote host.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include <future>
|
|
#include <grpcpp/server.h>
|
|
#include <grpcpp/server_builder.h>
|
|
#include <iostream>
|
|
#include <thread>
|
|
|
|
#include "Server.h"
|
|
|
|
using grpc::Server;
|
|
using grpc::ServerBuilder;
|
|
|
|
std::promise<void> ShutdownPromise;
|
|
|
|
int main() {
|
|
RPCConfig Config;
|
|
parseEnvironment(Config);
|
|
|
|
RemoteOffloadImpl Service(Config.MaxSize, Config.BlockSize);
|
|
|
|
ServerBuilder Builder;
|
|
Builder.AddListeningPort(Config.ServerAddresses[0],
|
|
grpc::InsecureServerCredentials());
|
|
Builder.RegisterService(&Service);
|
|
Builder.SetMaxMessageSize(INT_MAX);
|
|
std::unique_ptr<Server> Server(Builder.BuildAndStart());
|
|
if (getDebugLevel())
|
|
std::cerr << "Server listening on " << Config.ServerAddresses[0]
|
|
<< std::endl;
|
|
|
|
auto WaitForServer = [&]() { Server->Wait(); };
|
|
|
|
std::thread ServerThread(WaitForServer);
|
|
|
|
auto ShutdownFuture = ShutdownPromise.get_future();
|
|
ShutdownFuture.wait();
|
|
Server->Shutdown();
|
|
ServerThread.join();
|
|
|
|
return 0;
|
|
}
|