PR Category

Execute Infrastructure

PR Types

Bug fixes

Description

允许一个 UndefindVar 经过 loop body 之后变为多个 Value 的情况

def loop_with_inner_mutate_list(x):
    out = 100
    # a is an UndefinedVar
    for i in range(x):
        a = []
        a.append(x)
        a.append(x + 1)
        out += a[0]
        # After the loop, a is [x, x], which will be flattened to 2 elements
    return out

比如对于如上 case，最开始 a 是一个 UndefinedVar，经过 loop body 后变成了 [Value(), Value()]，这会导致 while op 的输入输出不匹配，因为如果输入是 UndefinedVar 我们会创建一个 FakeValue，而输出则是两个

因此新的方案对控制流进行了重写，在 build_while_op 时不再提前创建 FakeValue，而是在跑完 loop body 后，根据输出来 infer 输入里的 UndefinedVar 的 type，此时才创建 FakeValue

比如

• UndefindVar() -> Value(type1)，会将输入变为 Value(type1)
• UndefindVar() -> [Value(type1), Value(type2)]，会将输入变为 [Value(type1), Value(type2)]

新增的 FakeValue 会作为 PyWhileOp 的 extra_inputs_，会在 OptimizeUpdate 的时候创建新的 WhileOp，以更新输入输出

PCard-66972