File tree Expand file tree Collapse file tree 1 file changed +7
-1
lines changed Expand file tree Collapse file tree 1 file changed +7
-1
lines changed Original file line number Diff line number Diff line change @@ -531,14 +531,20 @@ def set_backend(name, unsafe=False):
531
531
If False, does not switch backend if current backend is not unified backend.
532
532
"""
533
533
if (backend .is_unified () == False and unsafe == False ):
534
- raise RuntimeError ("Can not change backend after loading %s" % name )
534
+ raise RuntimeError ("Can not change backend to %s after loading %s" % ( name , backend . name ()) )
535
535
536
536
if (backend .is_unified ()):
537
537
safe_call (backend .get ().af_set_backend (backend .get_id (name )))
538
538
else :
539
539
backend .set_unsafe (name )
540
540
return
541
541
542
+ def get_backend ():
543
+ """
544
+ Return the name of the backend
545
+ """
546
+ return backend .name ()
547
+
542
548
def get_backend_id (A ):
543
549
"""
544
550
Get backend name of an array
You can’t perform that action at this time.
0 commit comments