From 8cedcd3efe875f70376f63ca495efafecfffb477 Mon Sep 17 00:00:00 2001 From: feng_shuai Date: Tue, 15 Feb 2022 10:40:58 +0800 Subject: [PATCH] delete mish_convert_ut skip (#39432) --- paddle/fluid/inference/tensorrt/op_teller.cc | 7 ------- .../unittests/ir/inference/test_trt_convert_mish.py | 11 ----------- 2 files changed, 18 deletions(-) diff --git a/paddle/fluid/inference/tensorrt/op_teller.cc b/paddle/fluid/inference/tensorrt/op_teller.cc index 4c8d9d5096..767672007d 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 d223fd529a..d2b6924a9e 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() -- GitLab