diff --git a/test/ir/inference/CMakeLists.txt b/test/ir/inference/CMakeLists.txt index 21072869084886..bef3c83e8a0e10 100755 --- a/test/ir/inference/CMakeLists.txt +++ b/test/ir/inference/CMakeLists.txt @@ -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") @@ -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) diff --git a/test/ir/inference/test_mkldnn_shape_op.py b/test/ir/inference/test_onednn_shape_op.py similarity index 100% rename from test/ir/inference/test_mkldnn_shape_op.py rename to test/ir/inference/test_onednn_shape_op.py diff --git a/test/ir/inference/test_mkldnn_shuffle_channel_op.py b/test/ir/inference/test_onednn_shuffle_channel_op.py similarity index 100% rename from test/ir/inference/test_mkldnn_shuffle_channel_op.py rename to test/ir/inference/test_onednn_shuffle_channel_op.py diff --git a/tools/final_ut_parallel_rule.py b/tools/final_ut_parallel_rule.py index d618624030c79c..b69268052c0f0b 100644 --- a/tools/final_ut_parallel_rule.py +++ b/tools/final_ut_parallel_rule.py @@ -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',