wehub-resource-sync
36 lines
2.0 KiB
CMake
36 lines
2.0 KiB
CMake
if(NOT WIN32 AND TENSORRT_FOUND)
|
|
file(
|
|
GLOB TEST_OPS
|
|
RELATIVE "${CMAKE_CURRENT_SOURCE_DIR}"
|
|
"test_*.py")
|
|
string(REPLACE ".py" "" TEST_OPS "${TEST_OPS}")
|
|
|
|
foreach(TEST_OP ${TEST_OPS})
|
|
py_test_modules(${TEST_OP} MODULES ${TEST_OP})
|
|
endforeach()
|
|
set_tests_properties(test_converter_model_bert PROPERTIES TIMEOUT "500")
|
|
set_tests_properties(test_converter_model_dummy PROPERTIES TIMEOUT "500")
|
|
set_tests_properties(test_converter_model_resnet50 PROPERTIES TIMEOUT "500")
|
|
set_tests_properties(test_converter_model_resnet50_move PROPERTIES TIMEOUT
|
|
"500")
|
|
set_tests_properties(test_converter_conv PROPERTIES TIMEOUT "300")
|
|
set_tests_properties(test_export PROPERTIES TIMEOUT "500")
|
|
set_tests_properties(test_converter_norm PROPERTIES TIMEOUT "300")
|
|
set_tests_properties(test_converter_ops PROPERTIES TIMEOUT "600")
|
|
set_tests_properties(test_converter_stat PROPERTIES TIMEOUT "300")
|
|
set_tests_properties(test_converter_math PROPERTIES TIMEOUT "900")
|
|
set_tests_properties(test_converter_activation PROPERTIES TIMEOUT "900")
|
|
set_tests_properties(test_converter_others PROPERTIES TIMEOUT "300")
|
|
set_tests_properties(test_converter_manipulation PROPERTIES TIMEOUT "600")
|
|
set_tests_properties(test_converter_creation PROPERTIES TIMEOUT "300")
|
|
set_tests_properties(test_converter_attribute PROPERTIES TIMEOUT "300")
|
|
set_tests_properties(test_converter_common PROPERTIES TIMEOUT "500")
|
|
set_tests_properties(test_converter_input PROPERTIES TIMEOUT "500")
|
|
set_tests_properties(test_converter_vision PROPERTIES TIMEOUT "500")
|
|
set_tests_properties(test_converter_linalg PROPERTIES TIMEOUT "100")
|
|
set_tests_properties(test_converter_einsum PROPERTIES TIMEOUT "200")
|
|
set_tests_properties(test_converter_search PROPERTIES TIMEOUT "300")
|
|
set_tests_properties(test_converter_logic PROPERTIES TIMEOUT "300")
|
|
set_tests_properties(test_converter_pooling PROPERTIES TIMEOUT "300")
|
|
endif()
|