PR Category

Execute Infrastructure

PR Types

Bug fixes

Description

同 #71145 要解决的问题一样

def loop_update_iter_inner_normal(x):
    y = x + 1
    out = 0
    for item in y:
        y[0] = paddle.full([], 1, dtype="int64")
        out += y
    return out

对于上述代码，我们会转写类似如下的形式：

def loop_update_iter_inner_normal(x):
    y = x + 1
    out = 0
    __it = y
    __it_len = _jst.Len(__it)
    __it_idx = 0
    def cond():
        __it_idx < __it_len
    while cond():
        y[0] = paddle.full([], 1, dtype="int64")
        out += y
        __it_idx
    return out

值得注意的是，这里在 block 外直接使用了 __it 直接引用了 y，这导致了后续在 block 开始时 cond 里引用的 y 还是 block 外的 Value，而 y 同时又是写的变量，因此其作为了 while op 的输入

而 while instruction 有一个 early gc 的逻辑，在 while 执行开始时，会将后续用不到的 inputs 都 gc 掉，但实际上 while block 内是直接使用了 y 的外部变量的（非 block arg），而这里 early gc 的假设是如果一个 block body 使用了输入，那么一定使用的是 block arg 而不是外部变量，因而触发了 early gc 导致了后续使用外部变量时 holder 为空

#71145 尝试从转写层面解决该问题，但很难做到完善，因此本 PR 尝试使用 pass 来解决该问题

本 PR 修改 early gc 逻辑，当 loop body 内仍然有 user 时跳过 early gc

closes #71145

PCard-66972