diff --git a/paddle/fluid/eager/auto_code_generator/final_state_generator/eager_gen.py b/paddle/fluid/eager/auto_code_generator/final_state_generator/eager_gen.py index a601784042163ec698b9afe2284538f3c73d20e0..0f78763d6c959b1bfda61f80663334c680bf062c 100644 --- a/paddle/fluid/eager/auto_code_generator/final_state_generator/eager_gen.py +++ b/paddle/fluid/eager/auto_code_generator/final_state_generator/eager_gen.py @@ -1255,7 +1255,7 @@ class DygraphNodeGenerator(DygraphFunctionGeneratorBase): if num_outputs == 1: get_tensor_str = f"auto& {transformed_tensor_name} = grad_api_result;" else: - get_tensor_str = f"auto& {transformed_tensor_name} = grad_api_result[{fwd_position}];" + get_tensor_str = f"auto& {transformed_tensor_name} = grad_api_result[{grad_api_position}];" get_outputs_str += get_tensor_str + "\n" # Prepare for Node Creation if Necessary diff --git a/python/paddle/fluid/tests/unittests/test_inplace.py b/python/paddle/fluid/tests/unittests/test_inplace.py index 6670f2a174ecf8587bbb9cf1c4a67b9b1bcc0ad8..617e9811d630fb31fe28b041f66a6d26416ef3a9 100644 --- a/python/paddle/fluid/tests/unittests/test_inplace.py +++ b/python/paddle/fluid/tests/unittests/test_inplace.py @@ -325,8 +325,6 @@ class TestDygraphInplaceFlatten(TestDygraphInplace): return var.flatten_() -""" -# TODO: need to fix bug class TestDygraphInplaceScatter(TestDygraphInplace): def init_data(self): self.input_var_numpy = np.array([[1, 1], [2, 2], [3, 3]]) @@ -345,7 +343,6 @@ class TestDygraphInplaceScatter(TestDygraphInplace): [[1, 1], [2, 2], [3, 3], [4, 4]], dtype='float32') return paddle.scatter_(var, index, updates, overwrite=False) -""" class TestDygraphInplaceElu(TestDygraphInplace):