未验证 提交 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) ...@@ -65,6 +65,7 @@ function(inference_analysis_test_run TARGET)
"${multiValueArgs}" ${ARGN}) "${multiValueArgs}" ${ARGN})
inference_base_test_run(${TARGET} COMMAND ${analysis_test_COMMAND} ARGS inference_base_test_run(${TARGET} COMMAND ${analysis_test_COMMAND} ARGS
${analysis_test_ARGS}) ${analysis_test_ARGS})
set_tests_properties(${TARGET} PROPERTIES LABELS "RUN_TYPE=INFER")
endif() endif()
endfunction() endfunction()
...@@ -86,6 +87,7 @@ function(inference_analysis_test TARGET) ...@@ -86,6 +87,7 @@ function(inference_analysis_test TARGET)
${GLOB_PASS_LIB}) ${GLOB_PASS_LIB})
inference_base_test_run(${TARGET} COMMAND ${TARGET} ARGS inference_base_test_run(${TARGET} COMMAND ${TARGET} ARGS
${analysis_test_ARGS}) ${analysis_test_ARGS})
set_tests_properties(${TARGET} PROPERTIES LABELS "RUN_TYPE=INFER")
endif() endif()
endfunction() endfunction()
......
...@@ -12,6 +12,7 @@ endif() ...@@ -12,6 +12,7 @@ endif()
foreach(target ${TEST_IR_PASSES}) foreach(target ${TEST_IR_PASSES})
py_test_modules(${target} MODULES ${target}) py_test_modules(${target} MODULES ${target})
set_tests_properties(${target} PROPERTIES LABELS "RUN_TYPE=INFER")
endforeach() endforeach()
add_subdirectory(inference) add_subdirectory(inference)
......
...@@ -59,15 +59,18 @@ if(WITH_GPU AND TENSORRT_FOUND) ...@@ -59,15 +59,18 @@ if(WITH_GPU AND TENSORRT_FOUND)
"7.1") "7.1")
py_test_modules(${target} MODULES ${target}) py_test_modules(${target} MODULES ${target})
set_tests_properties(${target} PROPERTIES TIMEOUT 60) set_tests_properties(${target} PROPERTIES TIMEOUT 60)
set_tests_properties(${target} PROPERTIES LABELS "RUN_TYPE=INFER")
endif() endif()
else() else()
py_test_modules(${target} MODULES ${target}) py_test_modules(${target} MODULES ${target})
set_tests_properties(${target} PROPERTIES LABELS "RUN_TYPE=INFER")
endif() endif()
endforeach() endforeach()
foreach(target ${TEST_TRT_CONVERTER}) foreach(target ${TEST_TRT_CONVERTER})
py_test_modules(${target} MODULES ${target}) py_test_modules(${target} MODULES ${target})
set_tests_properties(${target} PROPERTIES TIMEOUT 300) set_tests_properties(${target} PROPERTIES TIMEOUT 300)
set_tests_properties(${target} PROPERTIES LABELS "RUN_TYPE=INFER")
endforeach() endforeach()
endif() endif()
...@@ -83,6 +86,7 @@ endforeach() ...@@ -83,6 +86,7 @@ endforeach()
if(WITH_MKLDNN) if(WITH_MKLDNN)
foreach(target ${TEST_MKLDNN_IR_PASSES}) foreach(target ${TEST_MKLDNN_IR_PASSES})
py_test_modules(${target} MODULES ${target}) py_test_modules(${target} MODULES ${target})
set_tests_properties(${target} PROPERTIES LABELS "RUN_TYPE=INFER")
endforeach() endforeach()
endif() endif()
...@@ -91,6 +95,7 @@ if(WITH_MKLDNN ...@@ -91,6 +95,7 @@ if(WITH_MKLDNN
AND WITH_GPU) AND WITH_GPU)
foreach(target ${TEST_INFERENCE_IR_PASSES}) foreach(target ${TEST_INFERENCE_IR_PASSES})
py_test_modules(${target} MODULES ${target}) py_test_modules(${target} MODULES ${target})
set_tests_properties(${target} PROPERTIES LABELS "RUN_TYPE=INFER")
endforeach() endforeach()
endif() endif()
...@@ -99,6 +104,7 @@ if(NOT WITH_MKLDNN ...@@ -99,6 +104,7 @@ if(NOT WITH_MKLDNN
AND NOT WITH_GPU) AND NOT WITH_GPU)
foreach(target ${TEST_INFERENCE_CPU_UT}) foreach(target ${TEST_INFERENCE_CPU_UT})
py_test_modules(${target} MODULES ${target}) py_test_modules(${target} MODULES ${target})
set_tests_properties(${target} PROPERTIES LABELS "RUN_TYPE=INFER")
endforeach() endforeach()
set_tests_properties(test_mul_lstm_fuse_pass PROPERTIES TIMEOUT 300) 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.
先完成此消息的编辑!
想要评论请 注册