1 //===-- Loader test to check the RPC interface with the loader ------------===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8
9 #include "include/llvm-libc-types/test_rpc_opcodes_t.h"
10 #include "src/__support/GPU/utils.h"
11 #include "src/__support/RPC/rpc_client.h"
12 #include "test/IntegrationTest/test.h"
13
14 using namespace LIBC_NAMESPACE;
15
test_add()16 static void test_add() {
17 uint64_t cnt = gpu::get_lane_id();
18 LIBC_NAMESPACE::rpc::Client::Port port =
19 LIBC_NAMESPACE::rpc::client.open<RPC_TEST_INCREMENT>();
20 port.send_and_recv(
21 [=](LIBC_NAMESPACE::rpc::Buffer *buffer, uint32_t) {
22 reinterpret_cast<uint64_t *>(buffer->data)[0] = cnt;
23 },
24 [&](LIBC_NAMESPACE::rpc::Buffer *buffer, uint32_t) {
25 cnt = reinterpret_cast<uint64_t *>(buffer->data)[0];
26 });
27 port.close();
28 EXPECT_EQ(cnt, gpu::get_lane_id() + 1);
29 EXPECT_EQ(gpu::get_thread_id(), gpu::get_lane_id());
30 }
31
TEST_MAIN(int argc,char ** argv,char ** envp)32 TEST_MAIN(int argc, char **argv, char **envp) {
33 test_add();
34
35 return 0;
36 }
37