xref: /llvm-project/libc/src/stdlib/gpu/malloc.cpp (revision b6bc9d72f65a5086f310f321e969d96e9a559e75)
1 //===-- GPU Implementation of malloc --------------------------------------===//
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 "src/stdlib/malloc.h"
10 #include "src/__support/RPC/rpc_client.h"
11 #include "src/__support/common.h"
12 
13 namespace LIBC_NAMESPACE {
14 
15 LLVM_LIBC_FUNCTION(void *, malloc, (size_t size)) {
16   void *ptr = nullptr;
17   rpc::Client::Port port = rpc::client.open<RPC_MALLOC>();
18   port.send_and_recv([=](rpc::Buffer *buffer) { buffer->data[0] = size; },
19                      [&](rpc::Buffer *buffer) {
20                        ptr = reinterpret_cast<void *>(buffer->data[0]);
21                      });
22   port.close();
23   return ptr;
24 }
25 
26 } // namespace LIBC_NAMESPACE
27