From a13c788a04b22a92104f7025a893bb4daa3d7a98 Mon Sep 17 00:00:00 2001 From: Qiao Longfei Date: Fri, 26 Oct 2018 21:51:41 +0800 Subject: [PATCH] fix a bug --- paddle/fluid/operators/lookup_table_op.cc | 2 +- python/paddle/fluid/transpiler/distribute_transpiler.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/paddle/fluid/operators/lookup_table_op.cc b/paddle/fluid/operators/lookup_table_op.cc index b9ac54e446..a4d1e812a5 100644 --- a/paddle/fluid/operators/lookup_table_op.cc +++ b/paddle/fluid/operators/lookup_table_op.cc @@ -115,7 +115,7 @@ class LookupTableOpGrad : public framework::OperatorWithKernel { protected: framework::OpKernelType GetExpectedKernelType( const framework::ExecutionContext& ctx) const override { - auto data_type = framework::GetDataTypeOfVar(ctx.InputVar("W")); + auto data_type = framework::GetDataTypeOfVar(ctx.InputVar("Out")); return framework::OpKernelType(data_type, ctx.device_context()); } }; diff --git a/python/paddle/fluid/transpiler/distribute_transpiler.py b/python/paddle/fluid/transpiler/distribute_transpiler.py index 5826db292b..b3a8958b22 100644 --- a/python/paddle/fluid/transpiler/distribute_transpiler.py +++ b/python/paddle/fluid/transpiler/distribute_transpiler.py @@ -487,7 +487,7 @@ class DistributeTranspiler(object): if init_op_num != 1: raise ValueError("table init op num should be 1, now is " + str( init_op_num)) - table_init_op = table_param_init_op[1] + table_init_op = table_param_init_op[0] self.startup_program.global_block().append_op( type="fake_init", inputs={}, -- GitLab