3#ifndef AWKWARD_KERNEL_DISPATCH_H_
4#define AWKWARD_KERNEL_DISPATCH_H_
38 const std::shared_ptr<LibraryPathCallback> &
callback);
43 std::map<kernel::lib, std::vector<std::shared_ptr<LibraryPathCallback>>> lib_path_callbacks;
44 std::mutex lib_path_callbacks_mutex;
107 (*awkward_free_fcn)(
reinterpret_cast<void const*
>(ptr));
125 template <
typename T>
144 const std::string&
indent,
145 const std::string&
pre,
146 const std::string&
post);
163 const std::string& cache_key);
170 template <
typename T>
175 return std::shared_ptr<T>(
184 return std::shared_ptr<T>(
189 throw std::runtime_error(
190 std::string(
"unrecognized ptr_lib in ptr_alloc<bool>"));
197 using UniquePtr = std::unique_ptr<T, UniquePtrDeleter<T>>;
199 template <
typename T>
209 throw std::runtime_error(
210 std::string(
"unrecognized ptr_lib in ptr_alloc<bool>"));
217 template <
typename T>
237 template <
typename T>
244 template <
typename T>
251 template <
typename T>
260 template <
typename T>
283 template <
typename T>
289 template <
typename ID>
407 template <
typename T>
417 template <
typename T>
428 template <
typename T>
441 template <
typename T>
449 template <
typename T>
458 template <
typename T>
471 template <
typename T>
485 template <
typename T>
566 template <
typename T>
573 template <
typename T>
582 template <
typename T>
591 template <
typename T>
619 const int8_t* originalmask,
632 template <
typename T>
641 template <
typename T>
650 template <
typename T>
657 template <
typename T,
typename I>
666 template <
typename T,
typename I>
692 template <
typename T>
711 template <
typename T>
744 template <
typename T>
754 template <
typename T>
760 template <
typename T>
813 template <
typename T>
825 template <
typename T>
831 template <
typename T>
839 template <
typename C,
typename T>
849 template <
typename C,
typename T>
861 template <
typename ID>
872 template <
typename C,
typename T>
884 template <
typename C,
typename T,
typename I>
897 template <
typename ID>
907 template <
typename T>
921 template <
typename T>
930 template <
typename T>
940 template <
typename T,
typename I>
950 template <
typename T,
typename I>
961 template <
typename T,
typename I>
973 template <
typename T>
981 template <
typename T>
990 template <
typename T>
1014 template <
typename T>
1024 template <
typename T>
1034 template <
typename T>
1043 template <
typename T>
1057 template <
typename T>
1065 template <
typename T>
1087 template <
typename T>
1094 template <
typename FROM,
typename TO>
1102 template <
typename TO>
1110 template <
typename FROM,
typename TO>
1118 template <
typename FROM,
typename TO>
1126 template <
typename TO>
1134 template <
typename FROM,
typename TO>
1143 template <
typename FROM,
typename TO>
1156 template <
typename FROM,
typename TO>
1168 template <
typename FROM,
typename TO>
1192 template <
typename FROM,
typename TO>
1213 template <
typename T,
typename I>
1229 template <
typename T,
typename I>
1245 template <
typename T,
typename I>
1261 template <
typename T,
typename I>
1273 template <
typename T>
1282 template <
typename T>
1291 template <
typename T,
typename I>
1300 template <
typename T>
1338 template <
typename T>
1346 template <
typename T>
1355 template <
typename T>
1366 template <
typename T>
1374 template <
typename T>
1383 template <
typename T>
1396 template <
typename T>
1409 template <
typename T>
1417 template <
typename T>
1428 template <
typename T>
1483 template <
typename IN>
1492 template <
typename OUT,
typename IN>
1501 template <
typename IN>
1510 template <
typename OUT,
typename IN>
1519 template <
typename IN>
1528 template <
typename OUT,
typename IN>
1538 template <
typename OUT,
typename IN>
1548 template <
typename OUT,
typename IN>
1557 template <
typename OUT,
typename IN>
1639 template <
typename T>
1649 template <
typename T>
1656 template <
typename T>
1740 template <
typename T>
1747 template <
typename T>
1758 template <
typename T>
1767 template <
typename T>
1776 template <
typename T>
1788 template <
typename T>
1802 template <
typename T>
1813 template <
typename T>
1837 template <
typename T>
1848 template <
typename T>
Filters, rearranges, and/or duplicates items in its content through an index, which has the same effe...
Definition IndexedArray.h:204
Definition kernel-dispatch.h:32
std::string awkward_library_path(kernel::lib ptr_lib)
void add_library_path_callback(kernel::lib ptr_lib, const std::shared_ptr< LibraryPathCallback > &callback)
Definition kernel-dispatch.h:26
virtual std::string library_path()=0
LibraryPathCallback()=default
Used as a std::shared_ptr deleter (second argument) to overload delete ptr with delete[] ptr.
Definition kernel-dispatch.h:74
void operator()(T const *ptr)
Called by std::shared_ptr when its reference count reaches zero.
Definition kernel-dispatch.h:78
Used as a std::shared_ptr deleter (second argument) to overload delete ptr with cudaFree.
Definition kernel-dispatch.h:98
void operator()(T const *ptr)
Called by std::shared_ptr when its reference count reaches zero.
Definition kernel-dispatch.h:102
Used as a std::shared_ptr deleter (second argument) to overload delete ptr with nothing (no dereferen...
Definition kernel-dispatch.h:126
void operator()(T const *ptr)
Called by std::shared_ptr when its reference count reaches zero.
Definition kernel-dispatch.h:130
#define LIBAWKWARD_EXPORT_SYMBOL
Definition common.h:45
#define ERROR
Definition common.h:28
int64_t len(const T &self)
Definition content.h:46
py::tuple identity(const T &self)
Creates a single identity as a Python tuple of integers and strings.
EXPORT_SYMBOL void * awkward_malloc(int64_t bytelength)
EXPORT_SYMBOL void awkward_free(void const *ptr)
ERROR ListArray_fill(kernel::lib ptr_lib, TO *tostarts, int64_t tostartsoffset, TO *tostops, int64_t tostopsoffset, const FROM *fromstarts, const FROM *fromstops, int64_t length, int64_t base)
ERROR UnionArray_flatten_combine_64(kernel::lib ptr_lib, int8_t *totags, int64_t *toindex, int64_t *tooffsets, const T *fromtags, const I *fromindex, int64_t length, int64_t **offsetsraws)
ERROR IndexedArray_simplify64_to64(kernel::lib ptr_lib, int64_t *toindex, const T *outerindex, int64_t outerlength, const int64_t *innerindex, int64_t innerlength)
ERROR localindex_64(kernel::lib ptr_lib, int64_t *toindex, int64_t length)
ERROR reduce_sum_64(kernel::lib ptr_lib, OUT *toptr, const IN *fromptr, const int64_t *parents, int64_t lenparents, int64_t outlength)
ERROR ListOffsetArray_toRegularArray(kernel::lib ptr_lib, int64_t *size, const T *fromoffsets, int64_t offsetslength)
ERROR Identities_extend(kernel::lib ptr_lib, ID *toptr, const ID *fromptr, int64_t fromlength, int64_t tolength)
ERROR ListArray_getitem_carry_64(kernel::lib ptr_lib, T *tostarts, T *tostops, const T *fromstarts, const T *fromstops, const int64_t *fromcarry, int64_t lenstarts, int64_t lencarry)
ERROR NumpyArray_quick_argsort(kernel::lib ptr_lib, int64_t *toptr, const T *fromptr, int64_t length, int64_t *tmpbeg, int64_t *tmpend, const int64_t *offsets, int64_t offsetslength, bool ascending, bool stable, int64_t maxlevels)
ERROR ByteMaskedArray_reduce_next_64(kernel::lib ptr_lib, int64_t *nextcarry, int64_t *nextparents, int64_t *outindex, const int8_t *mask, const int64_t *parents, int64_t length, bool validwhen)
ERROR Identities_from_RegularArray(kernel::lib ptr_lib, ID *toptr, const ID *fromptr, int64_t size, int64_t tolength, int64_t fromlength, int64_t fromwidth)
ERROR ListOffsetArray_reduce_nonlocal_outstartsstops_64(kernel::lib ptr_lib, int64_t *outstarts, int64_t *outstops, const int64_t *distincts, int64_t lendistincts, int64_t outlength)
ERROR ListOffsetArray_flatten_offsets_64(kernel::lib ptr_lib, int64_t *tooffsets, const T *outeroffsets, int64_t outeroffsetslen, const int64_t *inneroffsets, int64_t inneroffsetslen)
ERROR reduce_min_64(kernel::lib ptr_lib, OUT *toptr, const IN *fromptr, const int64_t *parents, int64_t lenparents, int64_t outlength, OUT identity)
ERROR NumpyArray_reduce_adjust_starts_64(kernel::lib ptr_lib, int64_t *toptr, int64_t outlength, const int64_t *parents, const int64_t *starts)
ERROR IndexedArray_flatten_none2empty_64(kernel::lib ptr_lib, int64_t *outoffsets, const T *outindex, int64_t outindexlength, const int64_t *offsets, int64_t offsetslength)
ERROR IndexedArray_simplify32_to64(kernel::lib ptr_lib, int64_t *toindex, const T *outerindex, int64_t outerlength, const int32_t *innerindex, int64_t innerlength)
ERROR ListArray_combinations_64(kernel::lib ptr_lib, int64_t **tocarry, int64_t *toindex, int64_t *fromindex, int64_t n, bool replacement, const T *starts, const T *stops, int64_t length)
ERROR IndexedArray_getitem_nextcarry_64(kernel::lib ptr_lib, int64_t *tocarry, const T *fromindex, int64_t lenindex, int64_t lencontent)
ERROR RegularArray_broadcast_tooffsets_size1_64(kernel::lib ptr_lib, int64_t *tocarry, const int64_t *fromoffsets, int64_t offsetslength)
ERROR Identities_to_Identities64(kernel::lib ptr_lib, int64_t *toptr, const T *fromptr, int64_t length, int64_t width)
ERROR RegularArray_getitem_next_range_64(kernel::lib ptr_lib, int64_t *tocarry, int64_t regular_start, int64_t step, int64_t len, int64_t size, int64_t nextsize)
ERROR NumpyArray_reduce_mask_ByteMaskedArray_64(kernel::lib ptr_lib, int8_t *toptr, const int64_t *parents, int64_t lenparents, int64_t outlength)
ERROR Identities_from_IndexedArray(kernel::lib ptr_lib, bool *uniquecontents, C *toptr, const C *fromptr, const T *fromindex, int64_t tolength, int64_t fromlength, int64_t fromwidth)
ERROR Index_carry_64(kernel::lib ptr_lib, T *toindex, const T *fromindex, const int64_t *carry, int64_t lenfromindex, int64_t length)
ERROR zero_mask8(kernel::lib ptr_lib, int8_t *tomask, int64_t length)
ERROR BitMaskedArray_to_IndexedOptionArray64(kernel::lib ptr_lib, int64_t *toindex, const uint8_t *frombitmask, int64_t bitmasklength, bool validwhen, bool lsb_order)
ERROR ListOffsetArray_reduce_nonlocal_nextstarts_64(kernel::lib ptr_lib, int64_t *nextstarts, const int64_t *nextparents, int64_t nextlen)
ERROR IndexedArray_reduce_next_nonlocal_nextshifts_64(kernel::lib ptr_lib, int64_t *nextshifts, const T *index, int64_t length)
ERROR ListArray_rpad_and_clip_length_axis1(kernel::lib ptr_lib, int64_t *tolength, const T *fromstarts, const T *fromstops, int64_t target, int64_t lenstarts)
ERROR Index_iscontiguous(kernel::lib ptr_lib, bool *result, const T *fromindex, int64_t length)
ERROR UnionArray_simplify8_64_to8_64(kernel::lib ptr_lib, int8_t *totags, int64_t *toindex, const T *outertags, const I *outerindex, const int8_t *innertags, const int64_t *innerindex, int64_t towhich, int64_t innerwhich, int64_t outerwhich, int64_t length, int64_t base)
ERROR RegularArray_combinations_64(kernel::lib ptr_lib, int64_t **tocarry, int64_t *toindex, int64_t *fromindex, int64_t n, bool replacement, int64_t size, int64_t length)
ERROR ListOffsetArray_compact_offsets_64(kernel::lib ptr_lib, int64_t *tooffsets, const T *fromoffsets, int64_t length)
ERROR ListArray_getitem_jagged_descend_64(kernel::lib ptr_lib, int64_t *tooffsets, const int64_t *slicestarts, const int64_t *slicestops, int64_t sliceouterlen, const T *fromstarts, const T *fromstops)
ERROR UnionArray_validity(kernel::lib ptr_lib, const T *tags, const I *index, int64_t length, int64_t numcontents, const int64_t *lencontents)
ERROR ByteMaskedArray_mask8(kernel::lib ptr_lib, int8_t *tomask, const int8_t *frommask, int64_t length, bool validwhen)
ERROR ListArray_getitem_next_array_advanced_64(kernel::lib ptr_lib, int64_t *tocarry, int64_t *toadvanced, const T *fromstarts, const T *fromstops, const int64_t *fromarray, const int64_t *fromadvanced, int64_t lenstarts, int64_t lenarray, int64_t lencontent)
ERROR IndexedArray_flatten_nextcarry_64(kernel::lib ptr_lib, int64_t *tocarry, const T *fromindex, int64_t lenindex, int64_t lencontent)
ERROR ListOffsetArray_local_preparenext_64(kernel::lib ptr_lib, int64_t *tocarry, const int64_t *fromindex, int64_t length)
ERROR ListArray_localindex_64(kernel::lib ptr_lib, int64_t *toindex, const T *offsets, int64_t length)
ERROR IndexedArray_fill_to64_count(kernel::lib ptr_lib, int64_t *toindex, int64_t toindexoffset, int64_t length, int64_t base)
ERROR Identities_from_ListOffsetArray(kernel::lib ptr_lib, C *toptr, const C *fromptr, const T *fromoffsets, int64_t tolength, int64_t fromlength, int64_t fromwidth)
ERROR ByteMaskedArray_overlay_mask8(kernel::lib ptr_lib, int8_t *tomask, const int8_t *theirmask, const int8_t *mymask, int64_t length, bool validwhen)
ERROR missing_repeat_64(kernel::lib ptr_lib, int64_t *outindex, const int64_t *index, int64_t indexlength, int64_t repetitions, int64_t regularsize)
ERROR RegularArray_getitem_next_at_64(kernel::lib ptr_lib, int64_t *tocarry, int64_t at, int64_t len, int64_t size)
ERROR NumpyArray_subrange_equal(kernel::lib ptr_lib, T *tmpptr, const int64_t *fromstarts, const int64_t *fromstops, int64_t length, bool *toequal)
ERROR NumpyArray_fill_tocomplex(kernel::lib ptr_lib, TO *toptr, int64_t tooffset, const FROM *fromptr, int64_t length)
ERROR IndexedArray_fill(kernel::lib ptr_lib, TO *toindex, int64_t toindexoffset, const FROM *fromindex, int64_t length, int64_t base)
ERROR UnionArray_fillindex(kernel::lib ptr_lib, TO *toindex, int64_t toindexoffset, const FROM *fromindex, int64_t length)
ERROR IndexedArray_reduce_next_64(kernel::lib ptr_lib, int64_t *nextcarry, int64_t *nextparents, int64_t *outindex, const T *index, int64_t *parents, int64_t length)
ERROR RegularArray_getitem_next_range_spreadadvanced_64(kernel::lib ptr_lib, int64_t *toadvanced, const int64_t *fromadvanced, int64_t len, int64_t nextsize)
ERROR Index_nones_as_index_64(kernel::lib ptr_lib, int64_t *toindex, int64_t length)
ERROR slicearray_ravel_64(kernel::lib ptr_lib, int64_t *toptr, const int64_t *fromptr, int64_t ndim, const int64_t *shape, const int64_t *strides)
ERROR ListOffsetArray_reduce_local_nextparents_64(kernel::lib ptr_lib, int64_t *nextparents, const int64_t *offsets, int64_t length)
ERROR ListOffsetArray_reduce_nonlocal_preparenext_64(kernel::lib ptr_lib, int64_t *nextcarry, int64_t *nextparents, int64_t nextlen, int64_t *maxnextparents, int64_t *distincts, int64_t distinctslen, int64_t *offsetscopy, const int64_t *offsets, int64_t length, const int64_t *parents, int64_t maxcount)
ERROR NumpyArray_argsort(kernel::lib ptr_lib, int64_t *toptr, const T *fromptr, int64_t length, const int64_t *offsets, int64_t offsetslength, bool ascending, bool stable)
ERROR NumpyArray_contiguous_init_64(kernel::lib ptr_lib, int64_t *toptr, int64_t skip, int64_t stride)
ERROR ListArray_getitem_next_array_64(kernel::lib ptr_lib, int64_t *tocarry, int64_t *toadvanced, const T *fromstarts, const T *fromstops, const int64_t *fromarray, int64_t lenstarts, int64_t lenarray, int64_t lencontent)
ERROR NumpyArray_fill_frombool(kernel::lib ptr_lib, TO *toptr, int64_t tooffset, const bool *fromptr, int64_t length)
ERROR RegularArray_getitem_carry_64(kernel::lib ptr_lib, int64_t *tocarry, const int64_t *fromcarry, int64_t lencarry, int64_t size)
ERROR reduce_prod_bool_64(kernel::lib ptr_lib, bool *toptr, const IN *fromptr, const int64_t *parents, int64_t lenparents, int64_t outlength)
ERROR NumpyArray_getitem_next_null_64(kernel::lib ptr_lib, uint8_t *toptr, const uint8_t *fromptr, int64_t len, int64_t stride, const int64_t *pos)
ERROR RegularArray_num_64(kernel::lib ptr_lib, int64_t *tonum, int64_t size, int64_t length)
ERROR reduce_max_64(kernel::lib ptr_lib, OUT *toptr, const IN *fromptr, const int64_t *parents, int64_t lenparents, int64_t outlength, OUT identity)
ERROR NumpyArray_contiguous_next_64(kernel::lib ptr_lib, int64_t *topos, const int64_t *frompos, int64_t len, int64_t skip, int64_t stride)
ERROR NumpyArray_copy(kernel::lib ptr_lib, uint8_t *toptr, const uint8_t *fromptr, int64_t len)
ERROR UnionArray_fillna_64(kernel::lib ptr_lib, int64_t *toindex, const T *fromindex, int64_t length)
ERROR IndexedArray_overlay_mask8_to64(kernel::lib ptr_lib, int64_t *toindex, const int8_t *mask, const T *fromindex, int64_t length)
ERROR UnionArray_filltags_to8_from8(kernel::lib ptr_lib, int8_t *totags, int64_t totagsoffset, const int8_t *fromtags, int64_t length, int64_t base)
ERROR NumpyArray_unique_strings(kernel::lib ptr_lib, T *toptr, const int64_t *offsets, int64_t offsetslength, int64_t *outoffsets, int64_t *tolength)
ERROR NumpyArray_getitem_next_array_advanced_64(kernel::lib ptr_lib, int64_t *nextcarryptr, const int64_t *carryptr, const int64_t *advancedptr, const int64_t *flatheadptr, int64_t lencarry, int64_t skip)
ERROR IndexedArray_index_of_nulls(kernel::lib ptr_lib, int64_t *toindex, const T *fromindex, int64_t lenindex, const int64_t *parents, const int64_t *starts)
void regularize_rangeslice(int64_t *start, int64_t *stop, bool posstep, bool hasstart, bool hasstop, int64_t length)
ERROR sorting_ranges(kernel::lib ptr_lib, int64_t *toindex, int64_t tolength, const int64_t *parents, int64_t parentslength)
const std::string lib_tostring(kernel::lib ptr_lib, void *ptr, const std::string &indent, const std::string &pre, const std::string &post)
Produces a <Lib> element for 'tostring' to indicate the kernel library.
ERROR IndexedArray_getitem_nextcarry_outindex_64(kernel::lib ptr_lib, int64_t *tocarry, T *toindex, const T *fromindex, int64_t lenindex, int64_t lencontent)
ERROR IndexedArray_ranges_next_64(kernel::lib ptr_lib, const T *index, const int64_t *fromstarts, const int64_t *fromstops, int64_t length, int64_t *tostarts, int64_t *tostops, int64_t *tolength)
ERROR ByteMaskedArray_toIndexedOptionArray64(kernel::lib ptr_lib, int64_t *toindex, const int8_t *mask, int64_t length, bool validwhen)
ERROR ListOffsetArray_rpad_and_clip_axis1_64(kernel::lib ptr_lib, int64_t *toindex, const T *fromoffsets, int64_t length, int64_t target)
UniquePtr< T > unique_malloc(kernel::lib ptr_lib, int64_t bytelength)
Definition kernel-dispatch.h:200
ERROR ListArray_getitem_jagged_apply_64(kernel::lib ptr_lib, int64_t *tooffsets, int64_t *tocarry, const int64_t *slicestarts, const int64_t *slicestops, int64_t sliceouterlen, const int64_t *sliceindex, int64_t sliceinnerlen, const T *fromstarts, const T *fromstops, int64_t contentlen)
ERROR IndexedArray_numnull(kernel::lib ptr_lib, int64_t *numnull, const T *fromindex, int64_t lenindex)
ERROR ByteMaskedArray_getitem_nextcarry_outindex_64(kernel::lib ptr_lib, int64_t *tocarry, int64_t *toindex, const int8_t *mask, int64_t length, bool validwhen)
ERROR ListOffsetArray_reduce_nonlocal_maxcount_offsetscopy_64(kernel::lib ptr_lib, int64_t *maxcount, int64_t *offsetscopy, const int64_t *offsets, int64_t length)
ERROR RegularArray_getitem_jagged_expand_64(kernel::lib ptr_lib, int64_t *multistarts, int64_t *multistops, const int64_t *singleoffsets, int64_t regularsize, int64_t regularlength)
ERROR ListArray_getitem_next_range_counts_64(kernel::lib ptr_lib, int64_t *total, const T *fromoffsets, int64_t lenstarts)
ERROR BitMaskedArray_to_ByteMaskedArray(kernel::lib ptr_lib, int8_t *tobytemask, const uint8_t *frombitmask, int64_t bitmasklength, bool validwhen, bool lsb_order)
ERROR ListArray_rpad_axis1_64(kernel::lib ptr_lib, int64_t *toindex, const T *fromstarts, const T *fromstops, T *tostarts, T *tostops, int64_t target, int64_t length)
ERROR ByteMaskedArray_numnull(kernel::lib ptr_lib, int64_t *numnull, const int8_t *mask, int64_t length, bool validwhen)
ERROR IndexedArray_ranges_carry_next_64(kernel::lib ptr_lib, const T *index, const int64_t *fromstarts, const int64_t *fromstops, int64_t length, int64_t *tocarry)
ERROR copy_to(kernel::lib to_lib, kernel::lib from_lib, void *to_ptr, void *from_ptr, int64_t bytelength)
Internal Function an array buffer from library FROM to library TO, usually between main memory and a ...
ERROR ListArray_num_64(kernel::lib ptr_lib, int64_t *tonum, const T *fromstarts, const T *fromstops, int64_t length)
ERROR NumpyArray_sort(kernel::lib ptr_lib, T *toptr, const T *fromptr, int64_t length, const int64_t *offsets, int64_t offsetslength, int64_t parentslength, bool ascending, bool stable)
std::shared_ptr< LibraryCallback > lib_callback
ERROR ListArray_validity(kernel::lib ptr_lib, const T *starts, const T *stops, int64_t length, int64_t lencontent)
const std::string fully_qualified_cache_key(kernel::lib ptr_lib, const std::string &cache_key)
FIXME.
ERROR reduce_prod_64(kernel::lib ptr_lib, OUT *toptr, const IN *fromptr, const int64_t *parents, int64_t lenparents, int64_t outlength)
ERROR reduce_count_64(kernel::lib ptr_lib, int64_t *toptr, const int64_t *parents, int64_t lenparents, int64_t outlength)
ERROR ListOffsetArray_argsort_strings(kernel::lib ptr_lib, int64_t *tocarry, const int64_t *fromparents, int64_t length, const uint8_t *stringdata, const int64_t *stringstarts, const int64_t *stringstops, bool is_stable, bool is_ascending, bool is_local)
ERROR ListOffsetArray_reduce_global_startstop_64(kernel::lib ptr_lib, int64_t *globalstart, int64_t *globalstop, const int64_t *offsets, int64_t length)
ERROR NumpyArray_getitem_boolean_nonzero_64(kernel::lib ptr_lib, int64_t *toptr, const int8_t *fromptr, int64_t length, int64_t stride)
ERROR NumpyArray_getitem_next_array_64(kernel::lib ptr_lib, int64_t *nextcarryptr, int64_t *nextadvancedptr, const int64_t *carryptr, const int64_t *flatheadptr, int64_t lencarry, int64_t lenflathead, int64_t skip)
ERROR Content_getitem_next_missing_jagged_getmaskstartstop(kernel::lib ptr_lib, int64_t *index_in, int64_t *offsets_in, int64_t *mask_out, int64_t *starts_out, int64_t *stops_out, int64_t length)
ERROR RegularArray_reduce_nonlocal_preparenext_64(kernel::lib ptr_lib, int64_t *nextcarry, int64_t *nextparents, const int64_t *parents, int64_t size, int64_t length)
ERROR ListOffsetArray_reduce_local_outoffsets_64(kernel::lib ptr_lib, int64_t *outoffsets, const int64_t *parents, int64_t lenparents, int64_t outlength)
ERROR index_rpad_and_clip_axis0_64(kernel::lib ptr_lib, int64_t *toindex, int64_t target, int64_t length)
ERROR UnionArray_filltags_to8_const(kernel::lib ptr_lib, int8_t *totags, int64_t totagsoffset, int64_t length, int64_t base)
ERROR ListArray_getitem_next_at_64(kernel::lib ptr_lib, int64_t *tocarry, const T *fromstarts, const T *fromstops, int64_t lenstarts, int64_t at)
ERROR NumpyArray_fill_tocomplex_frombool(kernel::lib ptr_lib, TO *toptr, int64_t tooffset, const bool *fromptr, int64_t length)
ERROR reduce_argmax_64(kernel::lib ptr_lib, OUT *toptr, const IN *fromptr, const int64_t *parents, int64_t lenparents, int64_t outlength)
ERROR ByteMaskedArray_reduce_next_nonlocal_nextshifts_64(kernel::lib ptr_lib, int64_t *nextshifts, const int8_t *mask, int64_t length, bool valid_when)
const int64_t lib_device_num(kernel::lib ptr_lib, void *ptr)
Returns the number of the device associated with the pointer.
ERROR ListArray_min_range(kernel::lib ptr_lib, int64_t *tomin, const T *fromstarts, const T *fromstops, int64_t lenstarts)
ERROR MaskedArray_getitem_next_jagged_project(kernel::lib ptr_lib, T *index, int64_t *starts_in, int64_t *stops_in, int64_t *starts_out, int64_t *stops_out, int64_t length)
ERROR ListArray_getitem_next_range_carrylength(kernel::lib ptr_lib, int64_t *carrylength, const T *fromstarts, const T *fromstops, int64_t lenstarts, int64_t start, int64_t stop, int64_t step)
ERROR NumpyArray_contiguous_copy_64(kernel::lib ptr_lib, uint8_t *toptr, const uint8_t *fromptr, int64_t len, int64_t stride, const int64_t *pos)
ERROR UnionArray_simplify_one_to8_64(kernel::lib ptr_lib, int8_t *totags, int64_t *toindex, const T *fromtags, const I *fromindex, int64_t towhich, int64_t fromwhich, int64_t length, int64_t base)
ERROR unique(kernel::lib ptr_lib, T *toptr, int64_t length, int64_t *tolength)
void * acquire_symbol(void *handle, const std::string &symbol_name)
Internal utility function to return an opaque ptr if an symbol is found for the corresponding handle....
ERROR carry_arange(kernel::lib ptr_lib, T *toptr, int64_t length)
ERROR UnionArray_simplify8_32_to8_64(kernel::lib ptr_lib, int8_t *totags, int64_t *toindex, const T *outertags, const I *outerindex, const int8_t *innertags, const int32_t *innerindex, int64_t towhich, int64_t innerwhich, int64_t outerwhich, int64_t length, int64_t base)
ERROR reduce_argmin_64(kernel::lib ptr_lib, OUT *toptr, const IN *fromptr, const int64_t *parents, int64_t lenparents, int64_t outlength)
ERROR RegularArray_getitem_next_array_64(kernel::lib ptr_lib, int64_t *tocarry, int64_t *toadvanced, const int64_t *fromarray, int64_t len, int64_t lenarray, int64_t size)
ERROR ListArray_broadcast_tooffsets_64(kernel::lib ptr_lib, int64_t *tocarry, const int64_t *fromoffsets, int64_t offsetslength, const T *fromstarts, const T *fromstops, int64_t lencontent)
ERROR NumpyArray_getitem_next_range_advanced_64(kernel::lib ptr_lib, int64_t *nextcarryptr, int64_t *nextadvancedptr, const int64_t *carryptr, const int64_t *advancedptr, int64_t lencarry, int64_t lenhead, int64_t skip, int64_t start, int64_t step)
ERROR one_mask8(kernel::lib ptr_lib, int8_t *tomask, int64_t length)
ERROR new_Identities(kernel::lib ptr_lib, T *toptr, int64_t length)
ERROR RegularArray_compact_offsets_64(kernel::lib ptr_lib, int64_t *tooffsets, int64_t length, int64_t size)
ERROR ListArray_compact_offsets_64(kernel::lib ptr_lib, int64_t *tooffsets, const T *fromstarts, const T *fromstops, int64_t length)
ERROR NumpyArray_getitem_next_at_64(kernel::lib ptr_lib, int64_t *nextcarryptr, const int64_t *carryptr, int64_t lencarry, int64_t skip, int64_t at)
ERROR Identities_getitem_carry_64(kernel::lib ptr_lib, ID *newidentitiesptr, const ID *identitiesptr, const int64_t *carryptr, int64_t lencarry, int64_t width, int64_t length)
ERROR RegularArray_rpad_and_clip_axis1_64(kernel::lib ptr_lib, int64_t *toindex, int64_t target, int64_t size, int64_t length)
ERROR slicemissing_check_same(kernel::lib ptr_lib, bool *same, const int8_t *bytemask, const int64_t *missingindex, int64_t length)
ERROR reduce_sum_bool_64(kernel::lib ptr_lib, bool *toptr, const IN *fromptr, const int64_t *parents, int64_t lenparents, int64_t outlength)
ERROR ByteMaskedArray_reduce_next_nonlocal_nextshifts_fromshifts_64(kernel::lib ptr_lib, int64_t *nextshifts, const int8_t *mask, int64_t length, bool valid_when, const int64_t *shifts)
std::unique_ptr< T, UniquePtrDeleter< T > > UniquePtr
Definition kernel-dispatch.h:197
ERROR ListArray_getitem_jagged_carrylen_64(kernel::lib ptr_lib, int64_t *carrylen, const int64_t *slicestarts, const int64_t *slicestops, int64_t sliceouterlen)
ERROR ListArray_getitem_jagged_shrink_64(kernel::lib ptr_lib, int64_t *tocarry, int64_t *tosmalloffsets, int64_t *tolargeoffsets, const int64_t *slicestarts, const int64_t *slicestops, int64_t length, const int64_t *missing)
ERROR IndexedArray_validity(kernel::lib ptr_lib, const T *index, int64_t length, int64_t lencontent, bool isoption)
ERROR RegularArray_getitem_next_array_advanced_64(kernel::lib ptr_lib, int64_t *tocarry, int64_t *toadvanced, const int64_t *fromadvanced, const int64_t *fromarray, int64_t len, int64_t lenarray, int64_t size)
ERROR ListArray_getitem_next_range_64(kernel::lib ptr_lib, T *tooffsets, int64_t *tocarry, const T *fromstarts, const T *fromstops, int64_t lenstarts, int64_t start, int64_t stop, int64_t step)
ERROR UnionArray_flatten_length_64(kernel::lib ptr_lib, int64_t *total_length, const T *fromtags, const I *fromindex, int64_t length, int64_t **offsetsraws)
ERROR ListArray_getitem_jagged_numvalid_64(kernel::lib ptr_lib, int64_t *numvalid, const int64_t *slicestarts, const int64_t *slicestops, int64_t length, const int64_t *missing, int64_t missinglength)
ERROR UnionArray_project_64(kernel::lib ptr_lib, int64_t *lenout, int64_t *tocarry, const T *fromtags, const I *fromindex, int64_t length, int64_t which)
ERROR IndexedArray_simplifyU32_to64(kernel::lib ptr_lib, int64_t *toindex, const T *outerindex, int64_t outerlength, const uint32_t *innerindex, int64_t innerlength)
ERROR NumpyArray_quick_sort(kernel::lib ptr_lib, T *tmpptr, int64_t *tmpbeg, int64_t *tmpend, const int64_t *fromstarts, const int64_t *fromstops, bool ascending, int64_t length, int64_t maxlevels)
ERROR Index_carry_nocheck_64(kernel::lib ptr_lib, T *toindex, const T *fromindex, const int64_t *carry, int64_t length)
ERROR IndexedArray_mask8(kernel::lib ptr_lib, int8_t *tomask, const T *fromindex, int64_t length)
ERROR NumpyArray_fill_scaled(kernel::lib ptr_lib, TO *toptr, int64_t tooffset, const FROM *fromptr, int64_t length, double scale)
T NumpyArray_getitem_at0(kernel::lib ptr_lib, T *ptr)
Internal utility kernel to avoid raw pointer access.
ERROR NumpyArray_contiguous_copy_from_many_64(kernel::lib ptr_lib, uint8_t *toptr, const uint8_t **fromptrs, int64_t *fromlens, int64_t len, int64_t stride, const int64_t *pos)
ERROR regularize_arrayslice_64(kernel::lib ptr_lib, int64_t *flatheadptr, int64_t lenflathead, int64_t length)
T index_getitem_at_nowrap(kernel::lib ptr_lib, T *ptr, int64_t at)
ERROR IndexedOptionArray_rpad_and_clip_mask_axis1_64(kernel::lib ptr_lib, int64_t *toindex, const int8_t *frommask, int64_t length)
ERROR NumpyArray_sort_asstrings(kernel::lib ptr_lib, T *toptr, const T *fromptr, const int64_t *offsets, int64_t offsetslength, int64_t *outoffsets, bool ascending, bool stable)
ERROR RegularArray_reduce_local_nextparents_64(kernel::lib ptr_lib, int64_t *nextparents, int64_t size, int64_t length)
ERROR NumpyArray_rearrange_shifted(kernel::lib ptr_lib, TO *toptr, const FROM *fromshifts, int64_t length, const FROM *fromoffsets, int64_t offsetslength, const FROM *fromparents, int64_t parentslength, const FROM *fromstarts, int64_t startslength)
ERROR ListArray_getitem_next_range_spreadadvanced_64(kernel::lib ptr_lib, int64_t *toadvanced, const int64_t *fromadvanced, const T *fromoffsets, int64_t lenstarts)
ERROR content_reduce_zeroparents_64(kernel::lib ptr_lib, int64_t *toparents, int64_t length)
ERROR ByteMaskedArray_getitem_nextcarry_64(kernel::lib ptr_lib, int64_t *tocarry, const int8_t *mask, int64_t length, bool validwhen)
std::shared_ptr< T > malloc(kernel::lib ptr_lib, int64_t bytelength)
Internal Function to allocate an empty array of a given length with a given type. The bytelength para...
Definition kernel-dispatch.h:171
lib
Definition kernel-dispatch.h:20
ERROR ListOffsetArray_getitem_adjust_offsets_64(kernel::lib ptr_lib, int64_t *tooffsets, int64_t *tononzero, const int64_t *fromoffsets, int64_t length, const int64_t *nonzero, int64_t nonzerolength)
ERROR ListOffsetArray_rpad_axis1_64(kernel::lib ptr_lib, int64_t *toindex, const T *fromoffsets, int64_t fromlength, int64_t target)
ERROR NumpyArray_reduce_adjust_starts_shifts_64(kernel::lib ptr_lib, int64_t *toptr, int64_t outlength, const int64_t *parents, const int64_t *starts, const int64_t *shifts)
ERROR index_rpad_and_clip_axis1_64(kernel::lib ptr_lib, int64_t *tostarts, int64_t *tostops, int64_t target, int64_t length)
decltype(kernel::array_deleter< T >()) UniquePtrDeleter
Definition kernel-dispatch.h:195
ERROR ListOffsetArray_getitem_adjust_offsets_index_64(kernel::lib ptr_lib, int64_t *tooffsets, int64_t *tononzero, const int64_t *fromoffsets, int64_t length, const int64_t *index, int64_t indexlength, const int64_t *nonzero, int64_t nonzerolength, const int8_t *originalmask, int64_t masklength)
ERROR reduce_countnonzero_64(kernel::lib ptr_lib, int64_t *toptr, const IN *fromptr, const int64_t *parents, int64_t lenparents, int64_t outlength)
void * acquire_handle(kernel::lib ptr_lib)
Internal utility function to return an opaque ptr if an handle is acquired for the specified ptr_lib....
ERROR UnionArray_fillindex_count_64(kernel::lib ptr_lib, int64_t *toindex, int64_t toindexoffset, int64_t length)
ERROR UnionArray_nestedfill_tags_index_64(kernel::lib ptr_lib, T *totags, I *toindex, int64_t *tmpstarts, T tag, const int64_t *fromcounts, int64_t length)
ERROR ByteMaskedArray_getitem_carry_64(kernel::lib ptr_lib, int8_t *tomask, const int8_t *frommask, int64_t lenmask, const int64_t *fromcarry, int64_t lencarry)
ERROR sorting_ranges_length(kernel::lib ptr_lib, int64_t *tolength, const int64_t *parents, int64_t parentslength)
ERROR UnionArray_regular_index(kernel::lib ptr_lib, I *toindex, I *current, int64_t size, const T *fromtags, int64_t length)
ERROR RegularArray_localindex_64(kernel::lib ptr_lib, int64_t *toindex, int64_t size, int64_t length)
ERROR IndexedArray_reduce_next_fix_offsets_64(kernel::lib ptr_lib, int64_t *outoffsets, const int64_t *starts, int64_t startslength, int64_t outindexlength)
ERROR ListOffsetArray_reduce_nonlocal_nextshifts_64(kernel::lib ptr_lib, int64_t *nummissing, int64_t *missing, int64_t *nextshifts, const int64_t *offsets, int64_t length, const int64_t *starts, const int64_t *parents, int64_t maxcount, int64_t nextlen, const int64_t *nextcarry)
ERROR NumpyArray_fill_fromcomplex(kernel::lib ptr_lib, TO *toptr, int64_t tooffset, const FROM *fromptr, int64_t length)
ERROR Index_to_Index64(kernel::lib ptr_lib, int64_t *toptr, const T *fromptr, int64_t length)
ERROR ListArray_combinations_length_64(kernel::lib ptr_lib, int64_t *totallen, int64_t *tooffsets, int64_t n, bool replacement, const T *starts, const T *stops, int64_t length)
ERROR NumpyArray_getitem_boolean_numtrue(kernel::lib ptr_lib, int64_t *numtrue, const int8_t *fromptr, int64_t length, int64_t stride)
ERROR ListOffsetArray_rpad_length_axis1(kernel::lib ptr_lib, T *tooffsets, const T *fromoffsets, int64_t fromlength, int64_t target, int64_t *tolength)
ERROR NumpyArray_fill(kernel::lib ptr_lib, TO *toptr, int64_t tooffset, const FROM *fromptr, int64_t length)
ERROR RegularArray_broadcast_tooffsets_64(kernel::lib ptr_lib, const int64_t *fromoffsets, int64_t offsetslength, int64_t size)
ERROR IndexedArray_getitem_nextcarry_outindex_mask_64(kernel::lib ptr_lib, int64_t *tocarry, int64_t *toindex, const T *fromindex, int64_t lenindex, int64_t lencontent)
ERROR RegularArray_getitem_next_array_regularize_64(kernel::lib ptr_lib, int64_t *toarray, const int64_t *fromarray, int64_t lenarray, int64_t size)
ERROR UnionArray_simplify8_U32_to8_64(kernel::lib ptr_lib, int8_t *totags, int64_t *toindex, const T *outertags, const I *outerindex, const int8_t *innertags, const uint32_t *innerindex, int64_t towhich, int64_t innerwhich, int64_t outerwhich, int64_t length, int64_t base)
ERROR Identities_from_UnionArray(kernel::lib ptr_lib, bool *uniquecontents, C *toptr, const C *fromptr, const T *fromtags, const I *fromindex, int64_t tolength, int64_t fromlength, int64_t fromwidth, int64_t which)
ERROR combinations(kernel::lib ptr_lib, T *toindex, int64_t n, bool replacement, int64_t singlelen)
ERROR UnionArray_regular_index_getsize(kernel::lib ptr_lib, int64_t *size, const T *fromtags, int64_t length)
ERROR ListArray_getitem_jagged_expand_64(kernel::lib ptr_lib, int64_t *multistarts, int64_t *multistops, const int64_t *singleoffsets, int64_t *tocarry, const T *fromstarts, const T *fromstops, int64_t jaggedsize, int64_t length)
ERROR IndexedArray_getitem_carry_64(kernel::lib ptr_lib, T *toindex, const T *fromindex, const int64_t *fromcarry, int64_t lenindex, int64_t lencarry)
ERROR IndexedArray_reduce_next_nonlocal_nextshifts_fromshifts_64(kernel::lib ptr_lib, int64_t *nextshifts, const T *index, int64_t length, const int64_t *shifts)
ERROR IndexedArray_local_preparenext_64(kernel::lib ptr_lib, int64_t *tocarry, const int64_t *starts, const int64_t *parents, const int64_t parentslength, const int64_t *nextparents, const int64_t nextlen)
ERROR NumpyArray_getitem_next_range_64(kernel::lib ptr_lib, int64_t *nextcarryptr, const int64_t *carryptr, int64_t lencarry, int64_t lenhead, int64_t skip, int64_t start, int64_t step)
ERROR IndexedArray_getitem_adjust_outindex_64(kernel::lib ptr_lib, int8_t *tomask, int64_t *toindex, int64_t *tononzero, const int64_t *fromindex, int64_t fromindexlength, const int64_t *nonzero, int64_t nonzerolength)
void index_setitem_at_nowrap(kernel::lib ptr_lib, T *ptr, int64_t at, T value)
ERROR Identities_from_ListArray(kernel::lib ptr_lib, bool *uniquecontents, C *toptr, const C *fromptr, const T *fromstarts, const T *fromstops, int64_t tolength, int64_t fromlength, int64_t fromwidth)
Definition BitMaskedArray.h:15