chromium/third_party/xnnpack/src/src/xnnpack/allocator.h

// Copyright 2019 Google LLC
//
// This source code is licensed under the BSD-style license found in the
// LICENSE file in the root directory of this source tree.

#pragma once

#include <limits.h>
#include <stddef.h>
#include <stdlib.h>
#include <string.h>

#if defined(_MSC_VER)
  #include <malloc.h>
#elif !defined(__GNUC__)
  #include <alloca.h>
#endif

#include "xnnpack.h"
#include "xnnpack/common.h"
#include "xnnpack/params.h"

#ifdef __cplusplus
extern "C" {
#endif

XNN_INTERNAL extern const struct xnn_allocator xnn_default_allocator;

inline static void* xnn_allocate_memory(size_t memory_size) { … }

inline static void* xnn_allocate_zero_memory(size_t memory_size) { … }

inline static void* xnn_reallocate_memory(void* memory_pointer, size_t memory_size) { … }

inline static void xnn_release_memory(void* memory_pointer) { … }

inline static void* xnn_allocate_simd_memory(size_t memory_size) { … }

inline static void* xnn_allocate_zero_simd_memory(size_t memory_size) { … }

inline static void xnn_release_simd_memory(void* memory_pointer) { … }

#if defined(__GNUC__) && defined(__BIGGEST_ALIGNMENT__) && (__BIGGEST_ALIGNMENT__ >= XNN_ALLOCATION_ALIGNMENT)
  #define XNN_SIMD_ALLOCA …
#elif (defined(__clang_major__) && (__clang_major__ >= 4)) || \
    (defined(__GNUC__) && (__GNUC__ >= 5 || __GNUC__ == 4 && __GNUC_MINOR__ >= 7) && !defined(__INTEL_COMPILER))
  // Builtin expects alignment in bits.
  #define XNN_SIMD_ALLOCA(size) …
#elif defined(__GNUC__)
  #define XNN_SIMD_ALLOCA …
#elif defined(_MSC_VER)
  #define XNN_SIMD_ALLOCA …
#else
  #define XNN_SIMD_ALLOCA …
#endif

#ifdef __cplusplus
}  // extern "C"
#endif