diff --git a/paddle/fluid/inference/tensorrt/op_teller.cc b/paddle/fluid/inference/tensorrt/op_teller.cc index 4a632bef7743d8125aec191e4943233dedcedaa8..e73237ab13def6bf78bb991ab6df5ef4e5e80080 100644 --- a/paddle/fluid/inference/tensorrt/op_teller.cc +++ b/paddle/fluid/inference/tensorrt/op_teller.cc @@ -1327,20 +1327,6 @@ bool OpTeller::Tell(const framework::ir::Node* node, bool use_no_calib_int8, "the shuffle_channel op does not support dynamic shape yet"; return false; } - auto* block = desc.Block(); - if (block == nullptr) { - VLOG(3) << "The block desc is nullptr, we can't continue to analyze. " - "Developers need to check whether block_desc is passed in " - "the pass."; - return false; - } - auto* input_desc = block->FindVar(desc.Input("X").front()); - const auto input_shape = input_desc->GetShape(); - if (input_shape.size() != 4) { - VLOG(3) << "input dims is invalid. The input " - "dims size should be 4."; - return false; - } } if (op_type == "skip_layernorm") {