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.
54 lines
1.6 KiB
C++
54 lines
1.6 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;
|
|
|
|
static void test_add_simple() {
|
|
uint64_t num_additions =
|
|
10 + 10 * gpu::get_thread_id() + 10 * gpu::get_block_id();
|
|
uint64_t cnt = 0;
|
|
for (uint32_t i = 0; i < num_additions; ++i) {
|
|
LIBC_NAMESPACE::rpc::Client::Port port =
|
|
LIBC_NAMESPACE::rpc::client.open<LIBC_TEST_INCREMENT>();
|
|
port.send_and_recv(
|
|
[=](LIBC_NAMESPACE::rpc::Buffer *buffer, uint32_t) {
|
|
reinterpret_cast<uint64_t *>(buffer->data)[0] = cnt;
|
|
},
|
|
[&](LIBC_NAMESPACE::rpc::Buffer *buffer, uint32_t) {
|
|
cnt = reinterpret_cast<uint64_t *>(buffer->data)[0];
|
|
});
|
|
port.close();
|
|
}
|
|
ASSERT_TRUE(cnt == num_additions && "Incorrect sum");
|
|
}
|
|
|
|
// Test to ensure that the RPC mechanism doesn't hang on divergence.
|
|
static void test_noop(uint8_t data) {
|
|
LIBC_NAMESPACE::rpc::Client::Port port =
|
|
LIBC_NAMESPACE::rpc::client.open<LIBC_NOOP>();
|
|
port.send([=](LIBC_NAMESPACE::rpc::Buffer *buffer, uint32_t) {
|
|
buffer->data[0] = data;
|
|
});
|
|
port.close();
|
|
}
|
|
|
|
TEST_MAIN(int argc, char **argv, char **envp) {
|
|
test_add_simple();
|
|
|
|
if (gpu::get_thread_id() % 2)
|
|
test_noop(1);
|
|
else
|
|
test_noop(2);
|
|
|
|
return 0;
|
|
}
|