Skip to content

Commit 3c4bfcd

Browse files
committed
Modifying eval to take multiple parameters
1 parent 68c766d commit 3c4bfcd

File tree

1 file changed

+15
-3
lines changed

1 file changed

+15
-3
lines changed

arrayfire/device.py

Lines changed: 15 additions & 3 deletions
Original file line numberDiff line numberDiff 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

118128
def 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

0 commit comments

Comments
 (0)