def transform_varargs(annotator, v_func, v_shape, *data_v):
callspec = CallSpec.fromshape(v_shape.value, list(data_v))
v_vararg = callspec.w_stararg
if callspec.w_stararg:
s_vararg = annotator.annotation(callspec.w_stararg)
if not isinstance(s_vararg, SomeTuple):
raise AnnotatorError(
"Calls like f(..., *arg) require 'arg' to be a tuple")
n_items = len(s_vararg.items)
ops = [op.getitem(v_vararg, const(i)) for i in range(n_items)]
new_args = callspec.arguments_w + [hlop.result for hlop in ops]
if callspec.keywords:
newspec = CallSpec(new_args, callspec.keywords)
shape, data_v = newspec.flatten()
call_op = op.call_args(v_func, const(shape), *data_v)
else:
call_op = op.simple_call(v_func, *new_args)
ops.append(call_op)
return ops
def transform_varargs(annotator, v_func, v_shape, *data_v):
callspec = CallSpec.fromshape(v_shape.value, list(data_v))
v_vararg = callspec.w_stararg
if callspec.w_stararg:
s_vararg = annotator.annotation(callspec.w_stararg)
if not isinstance(s_vararg, SomeTuple):
raise AnnotatorError(
"Calls like f(..., *arg) require 'arg' to be a tuple")
n_items = len(s_vararg.items)
ops = [op.getitem(v_vararg, const(i)) for i in range(n_items)]
new_args = callspec.arguments_w + [hlop.result for hlop in ops]
if callspec.keywords:
newspec = CallSpec(new_args, callspec.keywords)
shape, data_v = newspec.flatten()
call_op = op.call_args(v_func, const(shape), *data_v)
else:
call_op = op.simple_call(v_func, *new_args)
ops.append(call_op)
return ops
