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
8 changes: 4 additions & 4 deletions test/ir/inference/CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -310,8 +310,8 @@ elseif(WITH_ONEDNN)
test_mkldnn_conv_transpose_bias_fuse_pass
test_mkldnn_conv3d_op
test_mkldnn_depthwise_conv_pass
test_mkldnn_shape_op
test_mkldnn_shuffle_channel_op)
test_onednn_shape_op
test_onednn_shuffle_channel_op)
foreach(target ${PIR_COVERAGE_MKLDNN_TESTS})
py_test_modules(${target}_pir MODULES ${target} ENVS FLAGS_enable_pir_api=1)
set_tests_properties(${target} PROPERTIES LABELS "RUN_TYPE=INFER")
Expand All @@ -334,8 +334,8 @@ elseif(WITH_ONEDNN)
message(STATUS "PIR Copied Test: ${target}_pir in inference test")
endforeach()

set_tests_properties(test_mkldnn_shape_op_pir PROPERTIES TIMEOUT 300)
set_tests_properties(test_mkldnn_shuffle_channel_op_pir PROPERTIES TIMEOUT
set_tests_properties(test_onednn_shape_op_pir PROPERTIES TIMEOUT 300)
set_tests_properties(test_onednn_shuffle_channel_op_pir PROPERTIES TIMEOUT
300)
set_tests_properties(test_onednn_conv_bias_fuse_pass_pir PROPERTIES TIMEOUT
300)
Expand Down
2 changes: 1 addition & 1 deletion tools/final_ut_parallel_rule.py
Original file line number Diff line number Diff line change
Expand Up @@ -52,7 +52,7 @@ def classify_cases_by_mem(rootPath):
'test_post_training_quantization_while',
'test_mkldnn_log_softmax_op',
'test_mkldnn_matmulv2_op',
'test_mkldnn_shape_op',
'test_onednn_shape_op',
'interceptor_pipeline_short_path_test',
'interceptor_pipeline_long_path_test',
'test_cpuonly_spawn',
Expand Down
Loading