fix bug in amp O2 (#32343)

4be3b057 · huangxu96 · GitHub · 7bae5e9a · 4be3b057
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

python/paddle/fluid/contrib/mixed_precision/fp16_utils.py python/paddle/fluid/contrib/mixed_precision/fp16_utils.py +1 -1

未找到文件。
--- a/python/paddle/fluid/contrib/mixed_precision/fp16_utils.py
+++ b/python/paddle/fluid/contrib/mixed_precision/fp16_utils.py
@@ -103,7 +103,7 @@ def _insert_cast_op(block, op, idx, src_dtype, dest_dtype):
            if in_name not in {'X', 'Z'}:
                continue
        for in_var_name in op.input(in_name):
-            in_var = block.var(in_var_name)
+            in_var = block._find_var_recursive(in_var_name)
            if in_var.type not in _valid_types or in_var.dtype == dest_dtype:
                continue
            if in_var.dtype == src_dtype: