File tree Expand file tree Collapse file tree 1 file changed +15
-3
lines changed
Expand file tree Collapse file tree 1 file changed +15
-3
lines changed Original file line number Diff line number Diff line change @@ -105,15 +105,25 @@ def sync(device=None):
105105 dev = device if device is not None else get_device ()
106106 safe_call (backend .get ().af_sync (dev ))
107107
108- def eval (A ):
108+ def __eval (* args ):
109+ for A in args :
110+ if isinstance (A , tuple ):
111+ __eval (* A )
112+ if isinstance (A , list ):
113+ __eval (* A )
114+ if isinstance (A , Array ):
115+ safe_call (backend .get ().af_eval (A .arr ))
116+
117+ def eval (* args ):
109118 """
110119 Evaluate the input
111120
112121 Parameters
113122 -----------
114- A : af.Array
123+ args : arguments to be evaluated
115124 """
116- safe_call (backend .get ().af_eval (A .arr ))
125+
126+ __eval (args )
117127
118128def device_mem_info ():
119129 """
@@ -200,3 +210,5 @@ def unlock_device_ptr(a):
200210 """
201211 ptr = ct .c_void_p (0 )
202212 safe_call (backend .get ().af_unlock_device_ptr (a .arr ))
213+
214+ from .array import Array
You can’t perform that action at this time.
0 commit comments