diff --git a/paddle/fluid/inference/tensorrt/op_teller.cc b/paddle/fluid/inference/tensorrt/op_teller.cc index 4c8d9d50965c069d4306eab29d71136eeb744639..767672007dfef67ecc4424fa6c962832599b0182 100644 --- a/paddle/fluid/inference/tensorrt/op_teller.cc +++ b/paddle/fluid/inference/tensorrt/op_teller.cc @@ -1247,13 +1247,6 @@ bool OpTeller::Tell(const framework::ir::Node* node, bool use_no_calib_int8, VLOG(3) << "mish op does not support input's dim is 1 in tensorrt."; return false; } - - if (!with_dynamic_shape) { - if (x_shape.size() == 2) { - VLOG(3) << "mish op does not support input's dim is 2 in tensorrt."; - return false; - } - } } if (op_type == "roi_align") { diff --git a/python/paddle/fluid/tests/unittests/ir/inference/test_trt_convert_mish.py b/python/paddle/fluid/tests/unittests/ir/inference/test_trt_convert_mish.py index d223fd529ab1741f156fd69e949a6f726fa5a589..d2b6924a9e938e6d7fad6c0930626d88da4f5123 100644 --- a/python/paddle/fluid/tests/unittests/ir/inference/test_trt_convert_mish.py +++ b/python/paddle/fluid/tests/unittests/ir/inference/test_trt_convert_mish.py @@ -154,17 +154,6 @@ class TrtConvertMishTest(TrtLayerAutoScanTest): self.add_skip_case(teller1, SkipReasons.TRT_NOT_SUPPORT, "Trt does not support 1-dimensional input.") - def teller2(program_config, predictor_config): - if (len(self.dynamic_shape.min_input_shape) == 0): - if self.dim1 != 0 and self.dim2 == 0 and self.dim3 == 0: - return True - return False - - self.add_skip_case( - teller2, SkipReasons.TRT_NOT_SUPPORT, - "Need to repair the case: the output of GPU and tensorrt has diff when the input dimension is 2 in static shape mode." - ) - def test(self): self.add_skip_trt_case() self.run_test()