Update

ysiraichi · ysiraichi · commit 476b1eaacc73 · 2025-04-26T08:37:21.000-03:00
[ghstack-poisoned]
diff --git a/torch/_tensor.py b/torch/_tensor.py
@@ -1696,7 +1696,8 @@ def __dlpack__(self, stream=None, max_version=None):
             None (default), PyTorch will fallback to DLPack 0.8.
         """
         if has_torch_function_unary(self):
-            return handle_torch_function(Tensor.__dlpack__, (self,), self, stream)
+            args = (self, stream, max_version)
+            return handle_torch_function(Tensor.__dlpack__, (self,), *args)
 
         # DLPack capsules can't capture all of PyTorch's semantics,
         # so we prohibit exporting tensors that would lose their properties like
diff --git a/torch/overrides.py b/torch/overrides.py
@@ -1511,7 +1511,7 @@ def get_testing_overrides() -> dict[Callable, Callable]:
         Tensor.view: lambda self, shape: -1,
         Tensor.view_as: lambda self, other: -1,
         Tensor.zero_: lambda self: -1,
-        Tensor.__dlpack__: lambda self, stream=None: -1,
+        Tensor.__dlpack__: lambda self, stream=None, max_version=None: -1,
         Tensor.__dlpack_device__: lambda self: -1,
         torch.linalg.lstsq: lambda self, b, cond=None, driver=None: -1,
     }  # fmt: skip
