bool + id的混合索引赋值在FLAGS_use_stride_kernel=0开关情况下行为不一致

[dynamicheart]

bug描述 Describe the Bug

复现代码：

import paddle
import numpy as np

position_ids = paddle.to_tensor(np.load('position_ids.npy')).reshape((3, 1, -1))
attention_mask = paddle.to_tensor(np.load('attention_mask.npy')).reshape((1, -1))
llm_positions = paddle.to_tensor(np.load('llm_positions.npy')).reshape((3, -1))

position_ids[..., 0, attention_mask[0] == 1] = llm_positions

print(position_ids)

输入数据：inputs.zip

FLAGS_use_stride_kernel=0输出：

Tensor(shape=[3, 1, 3985], dtype=int64, place=Place(gpu:0), stop_gradient=True,
       [[[1, 1, 1, ..., 1, 1, 1]],

        [[1, 1, 1, ..., 1, 1, 1]],

        [[1, 1, 1, ..., 1, 1, 1]]])

FLAGS_use_stride_kernel=1输出：

Tensor(shape=[3, 1, 3985], dtype=int64, place=Place(gpu:0), stop_gradient=True,
       [[[0 , 1 , 2 , ..., 44, 45, 46]],

        [[0 , 1 , 2 , ..., 44, 45, 46]],

        [[0 , 1 , 2 , ..., 44, 45, 46]]])

其他补充信息 Additional Supplementary Information

No response

[sunzhongkai588]

@LokeZhou 跟进一下