def make_expr_group(*exprs):
if len(exprs) == 1:
if isinstance(exprs[0], (list, tuple)):
exprs = exprs[0]
elif isinstance(exprs[0], ti.Matrix):
mat = exprs[0]
assert mat.m == 1
exprs = mat.entries
expr_group = _ti_core.ExprGroup()
for i in exprs:
if isinstance(i, ti.Matrix):
assert i.local_tensor_proxy is not None
expr_group.push_back(i.local_tensor_proxy)
else:
expr_group.push_back(Expr(i).ptr)
return expr_group
def make_expr_group(*exprs):
from taichi.lang.matrix import Matrix # pylint: disable=C0415
if len(exprs) == 1:
if isinstance(exprs[0], (list, tuple)):
exprs = exprs[0]
elif isinstance(exprs[0], Matrix):
mat = exprs[0]
assert mat.m == 1
exprs = mat.entries
expr_group = _ti_core.ExprGroup()
for i in exprs:
if isinstance(i, Matrix):
assert i.local_tensor_proxy is not None
expr_group.push_back(i.local_tensor_proxy)
else:
expr_group.push_back(Expr(i).ptr)
return expr_group