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

delete skip_case for dropout_ut (#39629)

上级 b8dbffb7
...@@ -112,7 +112,7 @@ class TrtConvertDropoutTest(TrtLayerAutoScanTest): ...@@ -112,7 +112,7 @@ class TrtConvertDropoutTest(TrtLayerAutoScanTest):
def generate_trt_nodes_num(attrs, dynamic_shape): def generate_trt_nodes_num(attrs, dynamic_shape):
if attrs[0]['dropout_implementation'] == "upscale_in_train": if attrs[0]['dropout_implementation'] == "upscale_in_train":
return 0, 2 return 0, 2
elif self.dims == 1: elif self.dims == 1 and dynamic_shape == False:
return 0, 3 return 0, 3
else: else:
return 1, 2 return 1, 2
...@@ -141,17 +141,7 @@ class TrtConvertDropoutTest(TrtLayerAutoScanTest): ...@@ -141,17 +141,7 @@ class TrtConvertDropoutTest(TrtLayerAutoScanTest):
True), 1e-5 True), 1e-5
def add_skip_trt_case(self): def add_skip_trt_case(self):
def teller1(program_config, predictor_config): pass
if len(
program_config.inputs['input_data'].shape
) == 2 and not predictor_config.tensorrt_dynamic_shape_enabled():
return True
return False
self.add_skip_case(
teller1, SkipReasons.TRT_NOT_IMPLEMENTED,
"The output shape has diff, but we can add shuffle layer to resolve it."
)
def test(self): def test(self):
self.add_skip_trt_case() self.add_skip_trt_case()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册