diff --git a/ivy/functional/ivy/device.py b/ivy/functional/ivy/device.py index d887aad881d5e..1aa1c82e7dc5e 100644 --- a/ivy/functional/ivy/device.py +++ b/ivy/functional/ivy/device.py @@ -164,7 +164,7 @@ def _shift_native_arrays_on_default_device(*args, **kwargs): ), [args, kwargs], ) - return args, kwargs, default_device + return args, kwargs, ivy.as_native_dev(default_device) # Device Queries #