From 89bfa9644e781619ef828f786748da23c4efe60d Mon Sep 17 00:00:00 2001
From: Zhang Zheng <32410583+ZzSean@users.noreply.github.com>
Date: Mon, 11 Apr 2022 11:23:53 +0800
Subject: [PATCH] Modify op-benchamrk script (#41470)

* Modify op-benchamrk script

* fix
---
 tools/ci_op_benchmark.sh | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tools/ci_op_benchmark.sh b/tools/ci_op_benchmark.sh
index 0937ebe5343..8e84eccc083 100644
--- a/tools/ci_op_benchmark.sh
+++ b/tools/ci_op_benchmark.sh
@@ -135,6 +135,8 @@ function load_CHANGE_OP_MAP {
   for change_file in ${CHANGE_OP_FILES[@]}
   do
     change_file_name=${change_file#*paddle/fluid/operators/}
+    change_file_name=${change_file_name#*paddle/phi/kernels/gpu/}
+    change_file_name=${change_file_name#*paddle/phi/kernels/gpudnn/}
     if [ -n "${PADDLE_FILENAME_OP_MAP[$change_file_name]}" ]
     then
       for op_name in ${PADDLE_FILENAME_OP_MAP[$change_file_name]}
-- 
GitLab