SymbolicScalar不支持循环内自增#
问题现象描述#
@pypto.frontend.jit
def add_kernel_1(a, b, c):
count = 0
for i in pypto.loop(20):
count = count + 1
当实际执行到i = 1时,count并不会像用户预期的那样从0依次增加到20。
可能原因#
当前PyPTO框架仅捕获了用户的Tensor操作,而未捕获用户的scalar操作,因此不会将count处理为变量。目前,只有循环变量能够实现自增。
处理步骤#
使用循环变量来表达自增逻辑。