libc/src/stdlib/gpu/malloc.cpp

   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