0b9d816b3e
Summary: This was originally kept separate so it didn't pollute the name space, but now I'm thinking it's just easier to bundle it in with the default interface. This means that we'll have a bit of extra code for people using the server.h file to handle libc opcodes, but it's minimal (3 functions) and it simplifies this. I'm doing this because I'm hoping to move the GPU tester binary to liboffload which handles `libc` opcodes internally except these. This is the easier option compared to adding a hook to register custom handlers there.
56 lines
1.6 KiB
C++
56 lines
1.6 KiB
C++
//===-- Common RPC server handler -----------------------------------------===//
|
|
//
|
|
// 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef LLVM_TOOLS_LLVM_GPU_LOADER_SERVER_H
|
|
#define LLVM_TOOLS_LLVM_GPU_LOADER_SERVER_H
|
|
|
|
#include <cstddef>
|
|
#include <cstdint>
|
|
|
|
#include "shared/rpc.h"
|
|
#include "shared/rpc_opcodes.h"
|
|
#include "shared/rpc_server.h"
|
|
|
|
template <uint32_t num_lanes, typename Alloc, typename Free>
|
|
inline uint32_t handle_server(rpc::Server &server, uint32_t index,
|
|
Alloc &&alloc, Free &&free) {
|
|
auto port = server.try_open(num_lanes, index);
|
|
if (!port)
|
|
return 0;
|
|
index = port->get_index() + 1;
|
|
|
|
int status = rpc::RPC_SUCCESS;
|
|
switch (port->get_opcode()) {
|
|
case LIBC_MALLOC: {
|
|
port->recv_and_send([&](rpc::Buffer *buffer, uint32_t) {
|
|
buffer->data[0] = reinterpret_cast<uintptr_t>(alloc(buffer->data[0]));
|
|
});
|
|
break;
|
|
}
|
|
case LIBC_FREE: {
|
|
port->recv([&](rpc::Buffer *buffer, uint32_t) {
|
|
free(reinterpret_cast<void *>(buffer->data[0]));
|
|
});
|
|
break;
|
|
}
|
|
default:
|
|
status = LIBC_NAMESPACE::shared::handle_libc_opcodes(*port, num_lanes);
|
|
break;
|
|
}
|
|
|
|
// Handle all of the `libc` specific opcodes.
|
|
if (status != rpc::RPC_SUCCESS)
|
|
handle_error("Error handling RPC server");
|
|
|
|
port->close();
|
|
|
|
return index;
|
|
}
|
|
|
|
#endif // LLVM_TOOLS_LLVM_GPU_LOADER_SERVER_H
|