DLPack (Capsule)

[ax3l]

I am trying to implement interfaces for DLPack in my projects, e.g., AMReX-Codes/pyamrex#454 using pybind11.

Centrally, this returns a PyCapsule object (pybind11: py::capsule): https://data-apis.org/array-api/latest/API_specification/generated/array_api.array.__dlpack__.html#array_api.array.__dlpack__

When implementing a pybind11 function that returns this type, pybind11-stubgen errors on:

pybind11_stubgen - [  ERROR] In amrex.space1d.amrex_1d_pybind.Array4_cdouble.__dlpack__ : Can't find/import 'capsule'

@sizmailov it looks like you tried to implement an annotation for this in #86 , but I see related reports like #240 that something might be amiss. Do you have any suggestions on how to implement this here?

[ax3l]

@sizmailov I found a fix and implemented it in #258 :)

[ax3l]

capsule solved with pybind11 3.0.0.

One still needs to register the DLDeviceType int32_t enum type if one implements the DLPack interface.