xref: /llvm-project/flang/runtime/memory.cpp (revision 3b337242ee165554f0017b00671381ec5b1ba855)
1 //===-- runtime/memory.cpp ------------------------------------------------===//
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 "flang/Runtime/memory.h"
10 #include "terminator.h"
11 #include "tools.h"
12 #include "flang/Runtime/freestanding-tools.h"
13 #include <cstdlib>
14 
15 namespace Fortran::runtime {
16 RT_OFFLOAD_API_GROUP_BEGIN
17 
AllocateMemoryOrCrash(const Terminator & terminator,std::size_t bytes)18 void *AllocateMemoryOrCrash(const Terminator &terminator, std::size_t bytes) {
19   if (void *p{std::malloc(bytes)}) {
20     return p;
21   }
22   if (bytes > 0) {
23     terminator.Crash(
24         "Fortran runtime internal error: out of memory, needed %zd bytes",
25         bytes);
26   }
27   return nullptr;
28 }
29 
ReallocateMemoryOrCrash(const Terminator & terminator,void * ptr,std::size_t newByteSize)30 void *ReallocateMemoryOrCrash(
31     const Terminator &terminator, void *ptr, std::size_t newByteSize) {
32   if (void *p{Fortran::runtime::realloc(ptr, newByteSize)}) {
33     return p;
34   }
35   if (newByteSize > 0) {
36     terminator.Crash("Fortran runtime internal error: memory realloc returned "
37                      "null, needed %zd bytes",
38         newByteSize);
39   }
40   return nullptr;
41 }
42 
FreeMemory(void * p)43 void FreeMemory(void *p) { std::free(p); }
44 
45 RT_OFFLOAD_API_GROUP_END
46 } // namespace Fortran::runtime
47