def foo(x: ti.i32, y: ti.i32, a: ti.template()):
a[0] = x + y
a[1] = x - y
a[2] = x * y
a[3] = ti.ti_float(x) / y
a[4] = x // y
a[5] = x % y
a[6] = x**y
a[7] = x << y
a[8] = x >> y
a[9] = x | y
a[10] = x ^ y
a[11] = x & y
def build_Call(ctx, node):
is_in_static_scope_prev = ctx.is_in_static_scope
if ASTTransformer.get_decorator(ctx, node) == 'static':
ctx.is_in_static_scope = True
build_stmt(ctx, node.func)
build_stmts(ctx, node.args)
build_stmts(ctx, node.keywords)
ctx.is_in_static_scope = is_in_static_scope_prev
args = []
for arg in node.args:
if isinstance(arg, ast.Starred):
args += arg.ptr
else:
args.append(arg.ptr)
keywords = dict(ChainMap(*[keyword.ptr for keyword in node.keywords]))
if isinstance(node.func, ast.Attribute):
attr_name = node.func.attr
if attr_name == 'format' and isinstance(node.func.value.ptr, str):
args.insert(0, node.func.value.ptr)
node.ptr = ti.ti_format(*args, **keywords)
else:
node.ptr = node.func.ptr(*args, **keywords)
elif isinstance(node.func, ast.Name):
func_name = node.func.id
if func_name == 'print':
node.ptr = ti.ti_print(*args, **keywords)
elif func_name == 'min':
node.ptr = ti.ti_min(*args, **keywords)
elif func_name == 'max':
node.ptr = ti.ti_max(*args, **keywords)
elif func_name == 'int':
node.ptr = ti.ti_int(*args, **keywords)
elif func_name == 'float':
node.ptr = ti.ti_float(*args, **keywords)
elif func_name == 'any':
node.ptr = ti.ti_any(*args, **keywords)
elif func_name == 'all':
node.ptr = ti.ti_all(*args, **keywords)
else:
node.ptr = node.func.ptr(*args, **keywords)
else:
node.ptr = node.func.ptr(*args, **keywords)
return node.ptr
def build_Call(ctx, node):
node.func = build_stmt(ctx, node.func)
node.args = build_stmts(ctx, node.args)
node.keywords = build_stmts(ctx, node.keywords)
args = []
for arg in node.args:
if isinstance(arg, ast.Starred):
args += arg.ptr
else:
args.append(arg.ptr)
keywords = dict(ChainMap(*[keyword.ptr for keyword in node.keywords]))
if isinstance(node.func, ast.Attribute):
attr_name = node.func.attr
if attr_name == 'format' and isinstance(node.func.value.ptr, str):
args.insert(0, node.func.value.ptr)
node.ptr = ti.ti_format(*args, **keywords)
else:
node.ptr = node.func.ptr(*args, **keywords)
elif isinstance(node.func, ast.Name):
func_name = node.func.id
if func_name == 'print':
node.ptr = ti.ti_print(*args, **keywords)
elif func_name == 'min':
node.ptr = ti.ti_min(*args, **keywords)
elif func_name == 'max':
node.ptr = ti.ti_max(*args, **keywords)
elif func_name == 'int':
node.ptr = ti.ti_int(*args, **keywords)
elif func_name == 'float':
node.ptr = ti.ti_float(*args, **keywords)
elif func_name == 'any':
node.ptr = ti.ti_any(*args, **keywords)
elif func_name == 'all':
node.ptr = ti.ti_all(*args, **keywords)
else:
node.ptr = node.func.ptr(*args, **keywords)
else:
node.ptr = node.func.ptr(*args, **keywords)
return node