未验证 提交 fea05f1f 编写于 作者: Y YUNSHEN XIE 提交者: GitHub

add labels for infer ut (#44279)

* add labels for infer ut

* add RUN_TYPE=INFER for cpp ut

* fix formaterror

* update
上级 5dfb87d9
......@@ -65,6 +65,7 @@ function(inference_analysis_test_run TARGET)
"${multiValueArgs}" ${ARGN})
inference_base_test_run(${TARGET} COMMAND ${analysis_test_COMMAND} ARGS
${analysis_test_ARGS})
set_tests_properties(${TARGET} PROPERTIES LABELS "RUN_TYPE=INFER")
endif()
endfunction()
......@@ -86,6 +87,7 @@ function(inference_analysis_test TARGET)
${GLOB_PASS_LIB})
inference_base_test_run(${TARGET} COMMAND ${TARGET} ARGS
${analysis_test_ARGS})
set_tests_properties(${TARGET} PROPERTIES LABELS "RUN_TYPE=INFER")
endif()
endfunction()
......
......@@ -12,6 +12,7 @@ endif()
foreach(target ${TEST_IR_PASSES})
py_test_modules(${target} MODULES ${target})
set_tests_properties(${target} PROPERTIES LABELS "RUN_TYPE=INFER")
endforeach()
add_subdirectory(inference)
......
......@@ -59,15 +59,18 @@ if(WITH_GPU AND TENSORRT_FOUND)
"7.1")
py_test_modules(${target} MODULES ${target})
set_tests_properties(${target} PROPERTIES TIMEOUT 60)
set_tests_properties(${target} PROPERTIES LABELS "RUN_TYPE=INFER")
endif()
else()
py_test_modules(${target} MODULES ${target})
set_tests_properties(${target} PROPERTIES LABELS "RUN_TYPE=INFER")
endif()
endforeach()
foreach(target ${TEST_TRT_CONVERTER})
py_test_modules(${target} MODULES ${target})
set_tests_properties(${target} PROPERTIES TIMEOUT 300)
set_tests_properties(${target} PROPERTIES LABELS "RUN_TYPE=INFER")
endforeach()
endif()
......@@ -83,6 +86,7 @@ endforeach()
if(WITH_MKLDNN)
foreach(target ${TEST_MKLDNN_IR_PASSES})
py_test_modules(${target} MODULES ${target})
set_tests_properties(${target} PROPERTIES LABELS "RUN_TYPE=INFER")
endforeach()
endif()
......@@ -91,6 +95,7 @@ if(WITH_MKLDNN
AND WITH_GPU)
foreach(target ${TEST_INFERENCE_IR_PASSES})
py_test_modules(${target} MODULES ${target})
set_tests_properties(${target} PROPERTIES LABELS "RUN_TYPE=INFER")
endforeach()
endif()
......@@ -99,6 +104,7 @@ if(NOT WITH_MKLDNN
AND NOT WITH_GPU)
foreach(target ${TEST_INFERENCE_CPU_UT})
py_test_modules(${target} MODULES ${target})
set_tests_properties(${target} PROPERTIES LABELS "RUN_TYPE=INFER")
endforeach()
set_tests_properties(test_mul_lstm_fuse_pass PROPERTIES TIMEOUT 300)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册