From d038b486dac1b8ad04bfaff90f99a5cae31a6449 Mon Sep 17 00:00:00 2001 From: Mahmoud Ashraf Date: Wed, 25 Oct 2023 23:21:39 +0300 Subject: [PATCH] fix: return `default_device` as native to be used directly in the backend --- ivy/functional/ivy/device.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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 #