From 4b9e748a12be698102c6f91fadcdcfd2a8942591 Mon Sep 17 00:00:00 2001 From: wenbin Date: Thu, 31 Mar 2022 10:03:59 +0800 Subject: [PATCH] remove shape check (#41143) --- paddle/fluid/inference/tensorrt/op_teller.cc | 14 -------------- 1 file changed, 14 deletions(-) diff --git a/paddle/fluid/inference/tensorrt/op_teller.cc b/paddle/fluid/inference/tensorrt/op_teller.cc index 4a632bef77..e73237ab13 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") { -- GitLab