[libc-commits] [PATCH] D148342: [libc] Add a test to directly stimulate the RPC interface

[Joseph Huber via Phabricator via libc-commits]

jhuber6 updated this revision to Diff 513900.
jhuber6 added a comment.

Addressing comments


Repository:
  rG LLVM Github Monorepo

CHANGES SINCE LAST ACTION
  https://reviews.llvm.org/D148342/new/

https://reviews.llvm.org/D148342

Files:
  libc/src/__support/RPC/rpc.h
  libc/test/integration/startup/gpu/CMakeLists.txt
  libc/test/integration/startup/gpu/rpc_test.cpp
  libc/utils/gpu/loader/Server.h


Index: libc/utils/gpu/loader/Server.h
===================================================================
--- libc/utils/gpu/loader/Server.h
+++ libc/utils/gpu/loader/Server.h
@@ -42,6 +42,11 @@
     });
     break;
   }
+  case __llvm_libc::rpc::Opcode::TEST_INCREMENT: {
+    port->recv_and_send(
+        [](__llvm_libc::rpc::Buffer *buffer) { buffer->data[0] += 1; });
+    break;
+  }
   default:
     port->recv([](__llvm_libc::rpc::Buffer *) { /* no-op */ });
     return;
Index: libc/test/integration/startup/gpu/rpc_test.cpp
===================================================================
--- /dev/null
+++ libc/test/integration/startup/gpu/rpc_test.cpp
@@ -0,0 +1,28 @@
+//===-- 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/RPC/rpc_client.h"
+#include "test/IntegrationTest/test.h"
+
+using namespace __llvm_libc;
+
+static void test_add_simple() {
+  constexpr int num_additions = 10000;
+  uint64_t cnt = 0;
+  for (int i = 0; i < num_additions; ++i) {
+    rpc::Port port = rpc::client.open(rpc::TEST_INCREMENT);
+    port.send_and_recv([=](rpc::Buffer *buffer) { buffer->data[0] = cnt; },
+                       [&](rpc::Buffer *buffer) { cnt = buffer->data[0]; });
+  }
+  ASSERT_TRUE(cnt == num_additions && "Incorrect sum");
+}
+
+TEST_MAIN(int argc, char **argv, char **envp) {
+  test_add_simple();
+  return 0;
+}
Index: libc/test/integration/startup/gpu/CMakeLists.txt
===================================================================
--- libc/test/integration/startup/gpu/CMakeLists.txt
+++ libc/test/integration/startup/gpu/CMakeLists.txt
@@ -12,3 +12,12 @@
     FRANCE=Paris
     GERMANY=Berlin
 )
+
+add_integration_test(
+  startup_rpc_test
+  SUITE libc-startup-tests
+  SRCS
+    rpc_test.cpp
+  DEPENDS
+    libc.src.__support.RPC.rpc_client
+)
Index: libc/src/__support/RPC/rpc.h
===================================================================
--- libc/src/__support/RPC/rpc.h
+++ libc/src/__support/RPC/rpc.h
@@ -33,6 +33,7 @@
   NOOP = 0,
   PRINT_TO_STDERR = 1,
   EXIT = 2,
+  TEST_INCREMENT = 3,
 };
 
 /// A fixed size channel used to communicate between the RPC client and server.


-------------- next part --------------
A non-text attachment was scrubbed...
Name: D148342.513900.patch
Type: text/x-patch
Size: 2478 bytes
Desc: not available
URL: <http://lists.llvm.org/pipermail/libc-commits/attachments/20230415/5e4258e5/attachment.bin>