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.
67 lines
2.1 KiB
C++
67 lines
2.1 KiB
C++
//===-- Loader test to check the RPC interface with the loader ------------===//
|
|
//
|
|
// 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "src/__support/GPU/utils.h"
|
|
#include "src/__support/RPC/rpc_client.h"
|
|
#include "test/IntegrationTest/test.h"
|
|
|
|
using namespace LIBC_NAMESPACE;
|
|
|
|
// Test to ensure that we can use aribtrary combinations of sends and recieves
|
|
// as long as they are mirrored.
|
|
static void test_interface(bool end_with_send) {
|
|
uint64_t cnt = 0;
|
|
LIBC_NAMESPACE::rpc::Client::Port port =
|
|
LIBC_NAMESPACE::rpc::client.open<LIBC_TEST_INTERFACE>();
|
|
port.send([&](LIBC_NAMESPACE::rpc::Buffer *buffer, uint32_t) {
|
|
buffer->data[0] = end_with_send;
|
|
});
|
|
port.send([&](LIBC_NAMESPACE::rpc::Buffer *buffer, uint32_t) {
|
|
buffer->data[0] = cnt = cnt + 1;
|
|
});
|
|
port.recv([&](LIBC_NAMESPACE::rpc::Buffer *buffer, uint32_t) {
|
|
cnt = buffer->data[0];
|
|
});
|
|
port.send([&](LIBC_NAMESPACE::rpc::Buffer *buffer, uint32_t) {
|
|
buffer->data[0] = cnt = cnt + 1;
|
|
});
|
|
port.recv([&](LIBC_NAMESPACE::rpc::Buffer *buffer, uint32_t) {
|
|
cnt = buffer->data[0];
|
|
});
|
|
port.send([&](LIBC_NAMESPACE::rpc::Buffer *buffer, uint32_t) {
|
|
buffer->data[0] = cnt = cnt + 1;
|
|
});
|
|
port.send([&](LIBC_NAMESPACE::rpc::Buffer *buffer, uint32_t) {
|
|
buffer->data[0] = cnt = cnt + 1;
|
|
});
|
|
port.recv([&](LIBC_NAMESPACE::rpc::Buffer *buffer, uint32_t) {
|
|
cnt = buffer->data[0];
|
|
});
|
|
port.recv([&](LIBC_NAMESPACE::rpc::Buffer *buffer, uint32_t) {
|
|
cnt = buffer->data[0];
|
|
});
|
|
if (end_with_send)
|
|
port.send([&](LIBC_NAMESPACE::rpc::Buffer *buffer, uint32_t) {
|
|
buffer->data[0] = cnt = cnt + 1;
|
|
});
|
|
else
|
|
port.recv([&](LIBC_NAMESPACE::rpc::Buffer *buffer, uint32_t) {
|
|
cnt = buffer->data[0];
|
|
});
|
|
port.close();
|
|
|
|
ASSERT_TRUE(cnt == 9 && "Invalid number of increments");
|
|
}
|
|
|
|
TEST_MAIN(int argc, char **argv, char **envp) {
|
|
test_interface(true);
|
|
test_interface(false);
|
|
|
|
return 0;
|
|
}
|