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

revert-22710-feature/integrated_ps_api
Wilber 5 years ago committed by GitHub
parent 903039a3c6
commit 870f465887

@ -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'
]

Loading…
Cancel
Save