Skip to content

Commit b3cea20

Browse files
authored
Merge pull request PaddlePaddle#33 from HuangShiqing/paddlebox
abacus-aibox-919 fix the partial_concat's shape error
2 parents bd08b0a + 3f840d1 commit b3cea20

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

paddle/fluid/operators/partial_concat_op_xpu.cc

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -63,7 +63,7 @@ class PartialConcatXPUKernel : public framework::OpKernel<T> {
6363

6464
std::vector<int> xshape = phi::vectorize<int>(in_vars[i]->dims());
6565
std::vector<int> starts = {0, start_index};
66-
std::vector<int> ends = {batch_size, start_index + partial_len + 1};//要截取的x的每个维度的终止坐标(不包含)
66+
std::vector<int> ends = {batch_size, start_index + partial_len};//要截取的x的每个维度的终止坐标(不包含)
6767

6868
int r = xpu::slice<T>(xpu_context,
6969
input_data,
@@ -129,7 +129,7 @@ class PartialConcatGradXPUKernel : public framework::OpKernel<T> {
129129

130130
std::vector<int> xshape = phi::vectorize<int>(out_grad->dims());
131131
std::vector<int> starts = {0, int(partial_len * i)};
132-
std::vector<int> ends = {batch_size, int(partial_len * i + partial_len + 1)};//要截取的x的每个维度的终止坐标(不包含)
132+
std::vector<int> ends = {batch_size, int(partial_len * i + partial_len)};//要截取的x的每个维度的终止坐标(不包含)
133133

134134
int r = xpu::slice<T>(xpu_context,
135135
out_grad_data,

0 commit comments

Comments
 (0)