未验证 提交 30bafff8 编写于 作者: C CtfGo 提交者: GitHub

fix temporary variable allocate (#36969)

上级 be2884eb
......@@ -197,7 +197,7 @@ class CinnLaunchOpKernel : public framework::OpKernel<T> {
if (!temp_variable_names.empty()) {
auto temp_cinn_tensors = details::GetCinnTensorsFromCompiledScope(
temp_variable_names, cinn_scope);
for (auto i = 0; i < output_variable_names.size(); ++i) {
for (auto i = 0; i < temp_variable_names.size(); ++i) {
const auto& var_name = temp_variable_names.at(i);
auto* tensor = temp_scope->Var(var_name)->GetMutable<LoDTensor>();
details::TensorMutableDataWithCinnInfo(place, temp_cinn_tensors.at(i),
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册