diff --git a/ge/hybrid/node_executor/aicpu/aicpu_node_executor.h b/ge/hybrid/node_executor/aicpu/aicpu_node_executor.h
index d5fe2949..2bc90eaa 100644
--- a/ge/hybrid/node_executor/aicpu/aicpu_node_executor.h
+++ b/ge/hybrid/node_executor/aicpu/aicpu_node_executor.h
@@ -17,9 +17,9 @@
 #ifndef GE_HYBRID_KERNEL_AICPU_NODE_EXECUTOR_H_
 #define GE_HYBRID_KERNEL_AICPU_NODE_EXECUTOR_H_
 
+#include "hybrid/node_executor/node_executor.h"
 #include "external/graph/types.h"
 #include "cce/aicpu_engine_struct.h"
-#include "hybrid/node_executor/node_executor.h"
 #include "aicpu_ext_info.h"
 
 namespace ge {
@@ -51,7 +51,7 @@ class AicpuNodeTaskBase : public NodeTask {
 
   virtual Status UpdateOutputShapeFromExtInfo(TaskContext &task_context);
 
-  Status UpdateShapeAndDataByResultSummary(TaskContext &context);
+  virtual Status UpdateShapeAndDataByResultSummary(TaskContext &context);
 
   Status InitForDependComputeTask(bool is_tfkernel);