未验证 提交 2eacef49 编写于 作者: C Chen Weihang 提交者: GitHub

fix saved expected kernel type error (#42743)

上级 b5e60db4
......@@ -317,9 +317,11 @@ PreparedOp PrepareImpl(
<< " | kernel key: " << pt_cpu_kernel_key
<< " | kernel: " << pt_cpu_kernel;
auto* cpu_ctx = pool.Get(paddle::platform::CPUPlace());
return PreparedOp(op, empty_ctx, expected_kernel_key, arg_map_fn,
default_kernel_signature, std::move(kernel_signature),
pt_cpu_kernel, cpu_ctx);
return PreparedOp(
op, empty_ctx,
framework::TransPhiKernelKeyToOpKernelType(pt_cpu_kernel_key),
arg_map_fn, default_kernel_signature, std::move(kernel_signature),
pt_cpu_kernel, cpu_ctx);
}
}
}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册