From d0dd5f4bca14cf4d1003166fb3bed2eae3630229 Mon Sep 17 00:00:00 2001
From: Illviljan <14371165+Illviljan@users.noreply.github.com>
Date: Tue, 20 Aug 2024 18:18:05 +0200
Subject: [PATCH] Update core.py

---
 xarray/namedarray/core.py | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/xarray/namedarray/core.py b/xarray/namedarray/core.py
index 9c514016ff4..570bf0955a2 100644
--- a/xarray/namedarray/core.py
+++ b/xarray/namedarray/core.py
@@ -50,6 +50,7 @@
 )
 
 if TYPE_CHECKING:
+    from enum import IntEnum
     from numpy.typing import NDArray
 
     from xarray.core.types import T_Chunks
@@ -631,6 +632,23 @@ def __index__(self, /) -> int:
     def __int__(self, /) -> int:
         return self._data.__int__()
 
+    # dlpack
+    def __dlpack__(
+        self,
+        /,
+        *,
+        stream: int | Any | None = None,
+        max_version: tuple[int, int] | None = None,
+        dl_device: tuple[IntEnum, int] | None = None,
+        copy: bool | None = None,
+    ) -> Any:
+        return self._data.__dlpack__(
+            stream=stream, max_version=max_version, dl_device=dl_device, copy=copy
+        )
+
+    def __dlpack_device__(self, /) -> tuple[IntEnum, int]:
+        return self._data.__dlpack_device__()
+
     # Arithmetic Operators
 
     def __neg__(self, /):