dlpack

Author: MarcoGorelli

pandas/core/interchange/buffer.py

@@ -100,6 +100,7 @@ def __init__(
             arr = chunked_array.combine_chunks()
         self._buffer = arr.buffers()[1]
         self._length = len(arr)
+        self._dlpack = arr.__dlpack__
 
     @property
     def bufsize(self) -> int:
@@ -119,7 +120,7 @@ def __dlpack__(self) -> Any:
         """
         Represent this structure as DLPack interface.
         """
-        return self._buffer.__dlpack__()
+        return self._dlpack()
 
     def __dlpack_device__(self) -> tuple[DlpackDeviceType, int | None]:
         """