From cb81826a6eaaf63d8ad07ebc39ad6790bc75e3a3 Mon Sep 17 00:00:00 2001 From: Pei Yang Date: Tue, 13 Apr 2021 20:41:57 +0800 Subject: [PATCH] extend multiclass_nms unittest timeout threshold (#32214) * extend multiclass_nms unittest timeout threshold * adjust timeout to 200s * temporarily disable multiclass_nms trt op teller --- paddle/fluid/inference/tensorrt/op_teller.cc | 1 - python/paddle/fluid/tests/unittests/ir/inference/CMakeLists.txt | 1 + 2 files changed, 1 insertion(+), 1 deletion(-) diff --git a/paddle/fluid/inference/tensorrt/op_teller.cc b/paddle/fluid/inference/tensorrt/op_teller.cc index b681b098c8c..179f534acef 100644 --- a/paddle/fluid/inference/tensorrt/op_teller.cc +++ b/paddle/fluid/inference/tensorrt/op_teller.cc @@ -114,7 +114,6 @@ struct SimpleOpTypeSetTeller : public Teller { "yolo_box", "roi_align", "affine_channel", - "multiclass_nms", "nearest_interp", "anchor_generator", }; diff --git a/python/paddle/fluid/tests/unittests/ir/inference/CMakeLists.txt b/python/paddle/fluid/tests/unittests/ir/inference/CMakeLists.txt index dfec1cc7572..3ebed017775 100644 --- a/python/paddle/fluid/tests/unittests/ir/inference/CMakeLists.txt +++ b/python/paddle/fluid/tests/unittests/ir/inference/CMakeLists.txt @@ -32,4 +32,5 @@ if(WITH_GPU AND TENSORRT_FOUND) set_tests_properties(test_trt_subgraph_pass PROPERTIES TIMEOUT 120) set_tests_properties(test_trt_activation_pass PROPERTIES TIMEOUT 120) set_tests_properties(test_trt_conv_pass PROPERTIES TIMEOUT 120) +set_tests_properties(test_trt_multiclass_nms_op PROPERTIES TIMEOUT 200) endif() -- GitLab