未验证 提交 332aadc0 编写于 作者: C ceci3 提交者: GitHub

fix quant pass (#49123)

上级 be4ea174
......@@ -1776,10 +1776,13 @@ class OutScaleForInferencePass:
in_node = graph._find_node_by_name(
op_node.outputs, var_name
)
if in_node.dtype() not in [
if (in_node.node.var() is None) or (
in_node.dtype()
not in [
core.VarDesc.VarType.FP64,
core.VarDesc.VarType.FP32,
]:
]
):
continue
scale_name = self._scale_name(var_name)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册