提交 870f4658 编写于 作者: W Wilber 提交者: GitHub

fix test_fusion_seqpool_concat lod level between compile and runtime (#22488)

上级 903039a3
......@@ -42,6 +42,12 @@ void FusionSeqPoolConcatOp::InferShape(
PADDLE_ENFORCE_EQ(ins_dims[0].size(), 2,
"The dims size of first input should be 2.");
ctx->SetOutputDim("Out", {-1, ins_dims[0][axis] * static_cast<int>(n)});
if (!ctx->IsRuntime()) {
// when compiling, the LodLevel of Out is set to be 1, which is consistent
// with that in running time.
ctx->SetLoDLevel("Out", 1);
}
}
framework::OpKernelType FusionSeqPoolConcatOp::GetExpectedKernelType(
......
......@@ -23,8 +23,7 @@ COMPILE_RUN_OP_WHITE_LIST = [
'sequence_reshape', 'generate_proposals', 'mine_hard_examples',
'retinanet_detection_output', 'ctc_align', 'fusion_seqpool_cvm_concat',
'gru', 'sequence_erase', 'rpn_target_assign', 'retinanet_target_assign',
'filter_by_instag', 'fusion_seqpool_concat', 'multiclass_nms',
'multiclass_nms2', 'im2sequence', 'generate_proposal_labels',
'distribute_fpn_proposals', 'detection_map', 'locality_aware_nms',
'var_conv_2d'
'filter_by_instag', 'multiclass_nms', 'multiclass_nms2', 'im2sequence',
'generate_proposal_labels', 'distribute_fpn_proposals', 'detection_map',
'locality_aware_nms', 'var_conv_2d'
]
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册