diff --git a/tools/check_op_benchmark_result.py b/tools/check_op_benchmark_result.py
index cef55f5ba0cd44076d5b56c00da5e23b0b320b30..9253604c9293e0f5a2954ee9aa447d220df33666 100644
--- a/tools/check_op_benchmark_result.py
+++ b/tools/check_op_benchmark_result.py
@@ -127,7 +127,7 @@ def update_api_info_file(fail_case_list, api_info_file):
     check_path_exists(api_info_file)
 
     # set of case names for performance check failures
-    fail_case_set = set(map(lambda x: x.split('_')[0], fail_case_list))
+    fail_case_set = set(map(lambda x: x.rsplit('_', 1)[0], fail_case_list))
 
     # list of api infos for performance check failures
     api_info_list = list()