From f8cac25f1b7c1cad97719f6f132355269d0f8808 Mon Sep 17 00:00:00 2001 From: wenbin Date: Fri, 9 Dec 2022 14:55:03 +0800 Subject: [PATCH] bug fix (#48829) --- .../fluid/tests/unittests/ir/inference/CMakeLists.txt | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/python/paddle/fluid/tests/unittests/ir/inference/CMakeLists.txt b/python/paddle/fluid/tests/unittests/ir/inference/CMakeLists.txt index ea4a5e0e5dd..05d26dde6ed 100755 --- a/python/paddle/fluid/tests/unittests/ir/inference/CMakeLists.txt +++ b/python/paddle/fluid/tests/unittests/ir/inference/CMakeLists.txt @@ -123,9 +123,7 @@ if(WITH_GPU AND TENSORRT_FOUND) #set_tests_properties(test_trt_multiclass_nms_op PROPERTIES TIMEOUT 200) set_tests_properties(test_trt_dynamic_shape PROPERTIES TIMEOUT 120) set_tests_properties(test_trt_inspector PROPERTIES TIMEOUT 60) - set_tests_properties(test_merge_layernorm_fuse_pass PROPERTIES TIMEOUT 180) - set_tests_properties(test_skip_merge_layernorm_fuse_pass PROPERTIES TIMEOUT - 180) + if(WITH_NV_JETSON) set_tests_properties( test_trt_pool_op @@ -159,6 +157,9 @@ if(WITH_GPU AND TENSORRT_FOUND) if(WITH_MKLDNN AND TENSORRT_FOUND AND WITH_GPU) + set_tests_properties(test_merge_layernorm_fuse_pass PROPERTIES TIMEOUT 180) + set_tests_properties(test_skip_merge_layernorm_fuse_pass PROPERTIES TIMEOUT + 180) set_tests_properties(test_emb_eltwise_layernorm_fuse_pass PROPERTIES TIMEOUT 120) -- GitLab