From f7cc94b90fc90fbece35d280b3c78ee7cbe4c8d4 Mon Sep 17 00:00:00 2001 From: Matt Haberland Date: Sun, 8 Nov 2026 10:03:48 -0800 Subject: [PATCH] Make torch.__array_namespace_info__().default_device() return default device --- src/array_api_compat/torch/_info.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/array_api_compat/torch/_info.py b/src/array_api_compat/torch/_info.py index fba69f00..2a771e68 100644 --- a/src/array_api_compat/torch/_info.py +++ b/src/array_api_compat/torch/_info.py @@ -120,7 +120,7 @@ def default_device(self): -------- https://github.com/data-apis/array-api/issues/835 """ - return torch.device("cpu") + return torch.get_default_device() def default_dtypes(self, *, device=None): """