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): """