21e92612c0
D97883 introduced a compile-time error in the experimental remote offloading libomptarget plugin, this patch fixes it and resolves a number of inconsistencies in the plugin as well: 1. Non-functional Asynchronous API 2. Unnecessarily verbose debug printing 3. Misc. code clean ups This is not intended to make any functional changes to the plugin. Differential Revision: https://reviews.llvm.org/D105325
52 lines
1.4 KiB
C++
52 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() {
|
|
ClientManagerConfigTy 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;
|
|
}
|