Awkward Array: include/awkward/kernel-utils.h Source File

Loading...
Searching...
No Matches
kernel-utils.h
Go to the documentation of this file.
1// BSD 3-Clause License; see https://github.com/scikit-hep/awkward-1.0/blob/main/LICENSE
2
3#ifndef AWKWARD_KERNEL_UTILS_H_
4#define AWKWARD_KERNEL_UTILS_H_
5
6#include "common.h"
7
8extern "C" {
9 EXPORT_SYMBOL int8_t
10 awkward_Index8_getitem_at_nowrap(
11 const int8_t* ptr,
12 int64_t at);
13 EXPORT_SYMBOL uint8_t
14 awkward_IndexU8_getitem_at_nowrap(
15 const uint8_t* ptr,
16 int64_t at);
17 EXPORT_SYMBOL int32_t
18 awkward_Index32_getitem_at_nowrap(
19 const int32_t* ptr,
20 int64_t at);
21 EXPORT_SYMBOL uint32_t
22 awkward_IndexU32_getitem_at_nowrap(
23 const uint32_t* ptr,
24 int64_t at);
25 EXPORT_SYMBOL int64_t
26 awkward_Index64_getitem_at_nowrap(
27 const int64_t* ptr,
28 int64_t at);
29
30 EXPORT_SYMBOL void
31 awkward_Index8_setitem_at_nowrap(
32 int8_t* ptr,
33 int64_t at,
34 int8_t value);
35 EXPORT_SYMBOL void
36 awkward_IndexU8_setitem_at_nowrap(
37 uint8_t* ptr,
38 int64_t at,
39 uint8_t value);
40 EXPORT_SYMBOL void
41 awkward_Index32_setitem_at_nowrap(
42 int32_t* ptr,
43 int64_t at,
44 int32_t value);
45 EXPORT_SYMBOL void
46 awkward_IndexU32_setitem_at_nowrap(
47 uint32_t* ptr,
48 int64_t at,
49 uint32_t value);
50 EXPORT_SYMBOL void
51 awkward_Index64_setitem_at_nowrap(
52 int64_t* ptr,
53 int64_t at,
54 int64_t value);
55
56 EXPORT_SYMBOL bool
57 awkward_NumpyArraybool_getitem_at0(
58 const bool* ptr);
59 EXPORT_SYMBOL int8_t
60 awkward_NumpyArray8_getitem_at0(
61 const int8_t* ptr);
62 EXPORT_SYMBOL uint8_t
63 awkward_NumpyArrayU8_getitem_at0(
64 const uint8_t* ptr);
65 EXPORT_SYMBOL int16_t
66 awkward_NumpyArray16_getitem_at0(
67 const int16_t* ptr);
68 EXPORT_SYMBOL uint16_t
69 awkward_NumpyArrayU16_getitem_at0(
70 const uint16_t* ptr);
71 EXPORT_SYMBOL int32_t
72 awkward_NumpyArray32_getitem_at0(
73 const int32_t* ptr);
74 EXPORT_SYMBOL uint32_t
75 awkward_NumpyArrayU32_getitem_at0(
76 const uint32_t* ptr);
77 EXPORT_SYMBOL int64_t
78 awkward_NumpyArray64_getitem_at0(
79 const int64_t* ptr);
80 EXPORT_SYMBOL uint64_t
81 awkward_NumpyArrayU64_getitem_at0(
82 const uint64_t* ptr);
83 EXPORT_SYMBOL float
84 awkward_NumpyArrayfloat32_getitem_at0(
85 const float* ptr);
86 EXPORT_SYMBOL double
87 awkward_NumpyArrayfloat64_getitem_at0(
88 const double* ptr);
89
90 EXPORT_SYMBOL void
91 awkward_regularize_rangeslice(
92 int64_t* start,
93 int64_t* stop,
94 bool posstep,
95 bool hasstart,
96 bool hasstop,
97 int64_t length
98 );
99
100 EXPORT_SYMBOL void
101 awkward_ListArray_combinations_step_64(
102 int64_t** tocarry,
103 int64_t* toindex,
104 int64_t* fromindex,
105 int64_t j,
106 int64_t stop,
107 int64_t n,
108 bool replacement
109 );
110
111 EXPORT_SYMBOL void* awkward_malloc(int64_t bytelength);
112 EXPORT_SYMBOL void awkward_free(void const *ptr);
113
114}
115
116#endif // AWKWARD_KERNEL_UTILS_H_
common.h
EXPORT_SYMBOL
#define EXPORT_SYMBOL
Definition: common.h:25
awkward_NumpyArrayU16_getitem_at0
EXPORT_SYMBOL uint16_t awkward_NumpyArrayU16_getitem_at0(const uint16_t *ptr)
awkward_NumpyArrayU8_getitem_at0
EXPORT_SYMBOL uint8_t awkward_NumpyArrayU8_getitem_at0(const uint8_t *ptr)
awkward_Index8_setitem_at_nowrap
EXPORT_SYMBOL void awkward_Index8_setitem_at_nowrap(int8_t *ptr, int64_t at, int8_t value)
awkward_Index64_getitem_at_nowrap
EXPORT_SYMBOL int64_t awkward_Index64_getitem_at_nowrap(const int64_t *ptr, int64_t at)
awkward_IndexU8_setitem_at_nowrap
EXPORT_SYMBOL void awkward_IndexU8_setitem_at_nowrap(uint8_t *ptr, int64_t at, uint8_t value)
awkward_Index64_setitem_at_nowrap
EXPORT_SYMBOL void awkward_Index64_setitem_at_nowrap(int64_t *ptr, int64_t at, int64_t value)
awkward_malloc
EXPORT_SYMBOL void * awkward_malloc(int64_t bytelength)
awkward_IndexU8_getitem_at_nowrap
EXPORT_SYMBOL uint8_t awkward_IndexU8_getitem_at_nowrap(const uint8_t *ptr, int64_t at)
awkward_Index32_getitem_at_nowrap
EXPORT_SYMBOL int32_t awkward_Index32_getitem_at_nowrap(const int32_t *ptr, int64_t at)
awkward_free
EXPORT_SYMBOL void awkward_free(void const *ptr)
awkward_NumpyArray32_getitem_at0
EXPORT_SYMBOL int32_t awkward_NumpyArray32_getitem_at0(const int32_t *ptr)
awkward_ListArray_combinations_step_64
EXPORT_SYMBOL void awkward_ListArray_combinations_step_64(int64_t **tocarry, int64_t *toindex, int64_t *fromindex, int64_t j, int64_t stop, int64_t n, bool replacement)
awkward_IndexU32_setitem_at_nowrap
EXPORT_SYMBOL void awkward_IndexU32_setitem_at_nowrap(uint32_t *ptr, int64_t at, uint32_t value)
awkward_regularize_rangeslice
EXPORT_SYMBOL void awkward_regularize_rangeslice(int64_t *start, int64_t *stop, bool posstep, bool hasstart, bool hasstop, int64_t length)
awkward_NumpyArray8_getitem_at0
EXPORT_SYMBOL int8_t awkward_NumpyArray8_getitem_at0(const int8_t *ptr)
awkward_IndexU32_getitem_at_nowrap
EXPORT_SYMBOL uint32_t awkward_IndexU32_getitem_at_nowrap(const uint32_t *ptr, int64_t at)
awkward_NumpyArrayfloat32_getitem_at0
EXPORT_SYMBOL float awkward_NumpyArrayfloat32_getitem_at0(const float *ptr)
awkward_NumpyArrayU64_getitem_at0
EXPORT_SYMBOL uint64_t awkward_NumpyArrayU64_getitem_at0(const uint64_t *ptr)
awkward_NumpyArray64_getitem_at0
EXPORT_SYMBOL int64_t awkward_NumpyArray64_getitem_at0(const int64_t *ptr)
awkward_NumpyArrayfloat64_getitem_at0
EXPORT_SYMBOL double awkward_NumpyArrayfloat64_getitem_at0(const double *ptr)
awkward_NumpyArrayU32_getitem_at0
EXPORT_SYMBOL uint32_t awkward_NumpyArrayU32_getitem_at0(const uint32_t *ptr)
awkward_NumpyArraybool_getitem_at0
EXPORT_SYMBOL bool awkward_NumpyArraybool_getitem_at0(const bool *ptr)
awkward_Index8_getitem_at_nowrap
EXPORT_SYMBOL int8_t awkward_Index8_getitem_at_nowrap(const int8_t *ptr, int64_t at)
awkward_NumpyArray16_getitem_at0
EXPORT_SYMBOL int16_t awkward_NumpyArray16_getitem_at0(const int16_t *ptr)
awkward_Index32_setitem_at_nowrap
EXPORT_SYMBOL void awkward_Index32_setitem_at_nowrap(int32_t *ptr, int64_t at, int32_t value)