Loading...
Searching...
No Matches
dlpack_util.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 AWKWARDPY_DLPACK_UTIL_H_
4
#define AWKWARDPY_DLPACK_UTIL_H_
5
6
#include "
awkward/util.h
"
7
8
#include "dlpack/dlpack.h"
9
10
namespace
py = pybind11;
11
namespace
ak
=
awkward
;
12
13
namespace
awkward
{
14
namespace
dlpack {
15
DLDataType
16
data_type_dispatch
(ak::util::dtype
dt
);
17
18
DLContext
19
device_context_dispatch
(ak::kernel::lib ptr_lib,
void
* ptr);
20
21
void
22
dlpack_deleter
(
DLManagedTensor
*
tensor
);
23
24
void
25
pycapsule_deleter
(
PyObject
*
dltensor
);
26
}
27
}
28
29
#endif
awkward::IndexedArrayOf
Filters, rearranges, and/or duplicates items in its content through an index, which has the same effe...
Definition
IndexedArray.h:204
awkward::dlpack::device_context_dispatch
DLContext device_context_dispatch(ak::kernel::lib ptr_lib, void *ptr)
awkward::dlpack::dlpack_deleter
void dlpack_deleter(DLManagedTensor *tensor)
awkward::dlpack::pycapsule_deleter
void pycapsule_deleter(PyObject *dltensor)
awkward::dlpack::data_type_dispatch
DLDataType data_type_dispatch(ak::util::dtype dt)
awkward
Definition
BitMaskedArray.h:15
util.h
include
awkward
python
dlpack_util.h
Generated by
1.10.0