1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
|
/*========================== begin_copyright_notice ============================
Copyright (C) 2017-2021 Intel Corporation
SPDX-License-Identifier: MIT
============================= end_copyright_notice ===========================*/
#ifndef ALLOCATOR_H
#define ALLOCATOR_H
#include <stdlib.h>
#include <stdint.h>
namespace IGC {
inline void *aligned_malloc(size_t nBytes, size_t alignBytes) {
void *kernel = nullptr;
#if defined(_WIN32)
kernel = _aligned_malloc(nBytes, alignBytes);
#elif defined(ANDROID)
kernel = memalign(alignBytes, nBytes);
#else // !defined(_WIN32) && !defined(ANDROID)
if (posix_memalign(&kernel, alignBytes, nBytes)) {
kernel = nullptr;
}
#endif
return kernel;
}
inline void aligned_free(void *ptr) {
#if defined(_WIN32)
_aligned_free(ptr);
#else // !defined(_WIN32)
free(ptr);
#endif
}
} // namespace IGC
#endif // ALLOCATOR_H
|
