From 20403fe909fe2c05c9d0b77368597d8605d01629 Mon Sep 17 00:00:00 2001 From: heliqi Date: Wed, 29 Dec 2021 11:16:24 +0800 Subject: [PATCH] add timeout for matmul_xx_fuse_pass (#38544) * del mkldnn options of baseline * add timeout for matmul_scale_fuse_pass * add timeout for matmul --- .../unittests/ir/inference/CMakeLists.txt | 18 +++++++++++++----- 1 file changed, 13 insertions(+), 5 deletions(-) diff --git a/python/paddle/fluid/tests/unittests/ir/inference/CMakeLists.txt b/python/paddle/fluid/tests/unittests/ir/inference/CMakeLists.txt index 1e05615a48b..bd2a8449446 100755 --- a/python/paddle/fluid/tests/unittests/ir/inference/CMakeLists.txt +++ b/python/paddle/fluid/tests/unittests/ir/inference/CMakeLists.txt @@ -98,14 +98,22 @@ if (WITH_MKLDNN AND TENSORRT_FOUND AND WITH_GPU) set_tests_properties(test_conv_act_mkldnn_fuse_pass PROPERTIES TIMEOUT 120) set_tests_properties(test_conv_elementwise_add2_act_fuse_pass PROPERTIES TIMEOUT 120) set_tests_properties(test_conv_elementwise_add_act_fuse_pass PROPERTIES TIMEOUT 120) - set_tests_properties(test_matmul_scale_fuse_pass PROPERTIES TIMEOUT 60) - set_tests_properties(test_matmul_v2_scale_fuse_pass PROPERTIES TIMEOUT 60) set_tests_properties(test_flatten2_matmul_fuse_pass PROPERTIES TIMEOUT 240) set_tests_properties(test_squeeze2_matmul_fuse_pass PROPERTIES TIMEOUT 240) set_tests_properties(test_reshape2_matmul_fuse_pass PROPERTIES TIMEOUT 240) - set_tests_properties(test_map_matmul_v2_to_matmul_pass PROPERTIES TIMEOUT 240) - set_tests_properties(test_map_matmul_v2_to_mul_pass PROPERTIES TIMEOUT 240) - set_tests_properties(test_map_matmul_to_mul_pass PROPERTIES TIMEOUT 240) + if (WIN32) + set_tests_properties(test_matmul_scale_fuse_pass PROPERTIES TIMEOUT 300) + set_tests_properties(test_matmul_v2_scale_fuse_pass PROPERTIES TIMEOUT 300) + set_tests_properties(test_map_matmul_v2_to_matmul_pass PROPERTIES TIMEOUT 360) + set_tests_properties(test_map_matmul_v2_to_mul_pass PROPERTIES TIMEOUT 360) + set_tests_properties(test_map_matmul_to_mul_pass PROPERTIES TIMEOUT 360) + else () + set_tests_properties(test_matmul_scale_fuse_pass PROPERTIES TIMEOUT 60) + set_tests_properties(test_matmul_v2_scale_fuse_pass PROPERTIES TIMEOUT 60) + set_tests_properties(test_map_matmul_v2_to_matmul_pass PROPERTIES TIMEOUT 120) + set_tests_properties(test_map_matmul_v2_to_mul_pass PROPERTIES TIMEOUT 120) + set_tests_properties(test_map_matmul_to_mul_pass PROPERTIES TIMEOUT 120) + endif() endif() if (WITH_MKLDNN) -- GitLab