diff --git a/paddle/fluid/framework/new_executor/interpretercore_util.cc b/paddle/fluid/framework/new_executor/interpretercore_util.cc index b135c214bd33c50b6d78251fad5bea42435d85e6..f7f2dd0213409f3f37a9ed2b376341aecb9bda48 100644 --- a/paddle/fluid/framework/new_executor/interpretercore_util.cc +++ b/paddle/fluid/framework/new_executor/interpretercore_util.cc @@ -289,7 +289,6 @@ std::tuple apply_place_transform_for_var( const OpKernelType& expected_kernel_key, const platform::Place& place, const std::string& var_name, const std::string& outer_name, const OpFuncNode& op_func_node, Variable* var, VariableScope* var_scope) { - auto& ins_name2id = op_func_node.input_index; auto& all_op_kernels = OperatorWithKernel::AllOpKernels(); platform::DeviceContextPool& pool = platform::DeviceContextPool::Instance(); std::string new_var_name = @@ -307,7 +306,7 @@ std::tuple apply_place_transform_for_var( : is_gpu_place(expected_kernel_key.place_) ? 1 : -1; std::map> copy_ins_name2id; - copy_ins_name2id["X"] = ins_name2id.at(outer_name); + copy_ins_name2id["X"] = {var_scope->VarId(var_name)}; std::map> copy_out_name2id; copy_out_name2id["Out"] = {var_scope->VarId(new_var_name)}; diff --git a/paddle/fluid/framework/new_executor/standalone_executor_test.cc b/paddle/fluid/framework/new_executor/standalone_executor_test.cc index 20bc5c7789d1267f74d7a26e9cce21205cb5a29e..3bb5121181fda74dc0575ca1e7a3bf89c76b6a5f 100644 --- a/paddle/fluid/framework/new_executor/standalone_executor_test.cc +++ b/paddle/fluid/framework/new_executor/standalone_executor_test.cc @@ -61,6 +61,8 @@ USE_OP(elementwise_max); USE_OP(elementwise_div); USE_OP(sgd); USE_OP(squared_l2_norm); +USE_OP(memcpy_h2d); +USE_OP(memcpy_d2h); paddle::framework::ProgramDesc load_from_file(const std::string& file_name) { std::ifstream fin(file_name, std::ios::in | std::ios::binary);