feat: set sub_backends by default if they are available (ivy-llc#25976)

iababio · Sep 27, 2023 · 89481e7 · 89481e7
1 parent a38e634
commit 89481e7
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/ivy/utils/backend/handler.py b/ivy/utils/backend/handler.py
@@ -418,7 +418,8 @@ def set_backend(backend: str, dynamic: bool = False):
 
         if dynamic:
             convert_from_numpy_to_target_backend(variable_ids, numpy_objs, devices)
-
+        for sub_backend in ivy.available_sub_backends:
+            ivy.set_sub_backend(sub_backend)
         if verbosity.level > 0:
             verbosity.cprint(f"backend stack: {backend_stack}")
     _handle_inplace_mode()

diff --git a/ivy/utils/backend/sub_backend_handler.py b/ivy/utils/backend/sub_backend_handler.py
@@ -278,15 +278,15 @@ def _unset_sub_backend_from_ivy(
 def clear_sub_backends():
     if ivy.current_sub_backends:
         ivy.__dict__.update(original_backend_dict)
-        ivy.current_sub_backends = []
+        ivy.current_sub_backends.clear()
 
 
 # This is only used in set_backend in handler.py
 def _clear_current_sub_backends():
     global original_backend_dict
     original_backend_dict = None
     if ivy.current_sub_backends:
-        ivy.current_sub_backends = []
+        ivy.current_sub_backends.clear()
 
 
 def find_available_sub_backends(sub_backends_loc):