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 11 #include "src/__support/GPU/allocator.h" 12 #include "src/__support/common.h" 13 #include "src/__support/macros/config.h" 14 15 namespace LIBC_NAMESPACE_DECL { 16 17 // FIXME: For now we just default to the NVIDIA device allocator which is 18 // always available on NVPTX targets. This will be implemented fully later. 19 #ifndef LIBC_TARGET_ARCH_IS_NVPTX 20 LLVM_LIBC_FUNCTION(void *, malloc, (size_t size)) { 21 return gpu::allocate(size); 22 } 23 #endif 24 25 } // namespace LIBC_NAMESPACE_DECL 26