未验证 提交 8cedcd3e 编写于 作者: F feng_shuai 提交者: GitHub

delete mish_convert_ut skip (#39432)

上级 7eb9593e
...@@ -1247,13 +1247,6 @@ bool OpTeller::Tell(const framework::ir::Node* node, bool use_no_calib_int8, ...@@ -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."; VLOG(3) << "mish op does not support input's dim is 1 in tensorrt.";
return false; 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") { if (op_type == "roi_align") {
......
...@@ -154,17 +154,6 @@ class TrtConvertMishTest(TrtLayerAutoScanTest): ...@@ -154,17 +154,6 @@ class TrtConvertMishTest(TrtLayerAutoScanTest):
self.add_skip_case(teller1, SkipReasons.TRT_NOT_SUPPORT, self.add_skip_case(teller1, SkipReasons.TRT_NOT_SUPPORT,
"Trt does not support 1-dimensional input.") "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): def test(self):
self.add_skip_trt_case() self.add_skip_trt_case()
self.run_test() self.run_test()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册