diff --git a/test/legacy_test/CMakeLists.txt b/test/legacy_test/CMakeLists.txt index 281ed6382b5f1d992724ba8db044048949ca9382..39eadc3af39aea1959b4b5db591263466a7c3b64 100644 --- a/test/legacy_test/CMakeLists.txt +++ b/test/legacy_test/CMakeLists.txt @@ -1213,10 +1213,14 @@ set(TEST_CINN_OPS test_assign_op test_flatten_contiguous_range_op) -foreach(TEST_CINN_OPS ${TEST_CINN_OPS}) +foreach(TEST_CINN_OP ${TEST_CINN_OPS}) if(WITH_CINN) - set_tests_properties(${TEST_CINN_OPS} PROPERTIES LABELS "RUN_TYPE=CINN") - set_tests_properties(${TEST_CINN_OPS} PROPERTIES TIMEOUT 200) + set_tests_properties(${TEST_CINN_OP} PROPERTIES LABELS "RUN_TYPE=CINN") + + get_test_property(${TEST_CINN_OP} TIMEOUT ORIGIN_TIME_OUT) + if((NOT ${ORIGIN_TIME_OUT}) OR (${ORIGIN_TIME_OUT} LESS 200)) + set_tests_properties(${TEST_CINN_OP} PROPERTIES TIMEOUT 200) + endif() endif() endforeach()