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
10namespace py = pybind11;
11namespace ak = awkward;
12
13namespace 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
DLContext device_context_dispatch(ak::kernel::lib ptr_lib, void *ptr)
void dlpack_deleter(DLManagedTensor *tensor)
void pycapsule_deleter(PyObject *dltensor)
DLDataType data_type_dispatch(ak::util::dtype dt)
Definition: BitMaskedArray.h:15