Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 2 additions & 0 deletions paddle/fluid/inference/analysis/CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -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()

Expand All @@ -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()

Expand Down
1 change: 1 addition & 0 deletions python/paddle/fluid/tests/unittests/ir/CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -52,15 +52,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()

Expand All @@ -76,6 +79,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()

Expand All @@ -84,6 +88,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()

Expand All @@ -92,6 +97,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)
Expand Down