Created by: githubutilities

scatter函数bug复现

import numpy as np
import paddle.fluid as fluid

def test_scatter_add():
    input = fluid.dygraph.to_variable(
        np.array([[1, 2],
         [5, 6]], dtype='float32'),
    )
    index = fluid.dygraph.to_variable(
        np.array([0, 1], dtype=np.int32)
    )
    updates = fluid.dygraph.to_variable(
        np.array([[3, 4],
         [3, 4]], dtype='float32'),
    )
    output = fluid.layers.scatter(input, index, updates, overwrite=False)
    print(output.numpy())

with fluid.dygraph.guard(fluid.CUDAPlace(0)):
    test_scatter_add()

输出 - CPU和GPU都能复现这个问题

[[3. 4.]
 [3. 4.]]

正确输出

[[4. 6.] [8. 10.]]

bug代码定位如下：

• CPU版代码 - https://github.com/PaddlePaddle/Paddle/blob/develop/paddle/fluid/operators/scatter.h#L134
• GPU版代码 - https://github.com/PaddlePaddle/Paddle/blob/develop/paddle/fluid/operators/scatter.cu.h#L99