From 1d84d94181c9357b1967ec5bea3f19fd7663e4ed Mon Sep 17 00:00:00 2001 From: zmxdream Date: Mon, 27 Nov 2023 14:01:08 +0800 Subject: [PATCH 1/2] fix pull_box_extend_sparse --- paddle/fluid/operators/pull_box_extended_sparse_op.h | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/paddle/fluid/operators/pull_box_extended_sparse_op.h b/paddle/fluid/operators/pull_box_extended_sparse_op.h index 459d9f4937024e..3807cf020abe0b 100644 --- a/paddle/fluid/operators/pull_box_extended_sparse_op.h +++ b/paddle/fluid/operators/pull_box_extended_sparse_op.h @@ -107,8 +107,8 @@ static void PullBoxExtendedSparseFunctor( reinterpret_cast(slot->data()); all_keys[i] = single_slot_keys; slot_lengths[i] = slot->numel(); - if (outputs[embedx_offset]->numel() == 0) { - outputs[embedx_offset]->set_layout(paddle::framework::DataLayout::UNDEFINED); + if (outputs[i]->numel() == 0) { + outputs[i]->set_layout(paddle::framework::DataLayout::UNDEFINED); } else { size_t offset = slot_dims0_offset[i] * dims1 * sizeof(T); total_values.set_offset(offset); @@ -116,8 +116,8 @@ static void PullBoxExtendedSparseFunctor( } auto *output = outputs[i]->mutable_data(ctx.GetPlace()); all_values[i] = reinterpret_cast(output); - if(outputs_extend[expand_offset]->numel()==0) { - outputs_extend[expand_offset]->set_layout(paddle::framework::DataLayout::UNDEFINED); + if(outputs_extend[i]->numel()==0) { + outputs_extend[i]->set_layout(paddle::framework::DataLayout::UNDEFINED); } else { size_t offset = slot_dims0_offset[i] * expand_dims1 * sizeof(T); total_values.set_offset(max_total_dims0 * dims1 * sizeof(T) + offset); From 715b0085493d314e1221a25edef05965aedfbf2a Mon Sep 17 00:00:00 2001 From: zmxdream Date: Mon, 27 Nov 2023 14:08:38 +0800 Subject: [PATCH 2/2] fix pull_box_extend_sparse --- paddle/fluid/operators/pull_box_extended_sparse_op.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/paddle/fluid/operators/pull_box_extended_sparse_op.h b/paddle/fluid/operators/pull_box_extended_sparse_op.h index 3807cf020abe0b..084aacf58c8729 100644 --- a/paddle/fluid/operators/pull_box_extended_sparse_op.h +++ b/paddle/fluid/operators/pull_box_extended_sparse_op.h @@ -116,7 +116,7 @@ static void PullBoxExtendedSparseFunctor( } auto *output = outputs[i]->mutable_data(ctx.GetPlace()); all_values[i] = reinterpret_cast(output); - if(outputs_extend[i]->numel()==0) { + if(outputs_extend[i]->numel() == 0) { outputs_extend[i]->set_layout(paddle::framework::DataLayout::UNDEFINED); } else { size_t offset = slot_dims0_offset[i] * expand_dims1 * sizeof(T);