fix ut

ge/hybrid/node_executor/aicpu/aicpu_node_executor.h

@@ -51,6 +51,8 @@ class AicpuNodeTaskBase : public NodeTask {
 
   virtual Status UpdateOutputShapeFromExtInfo(TaskContext &task_context);
 
+  virtual Status UpdateShapeAndDataByResultSummary(TaskContext &context) = 0;
+
   Status UpdateShapeToOutputDesc(TaskContext &task_context, const GeShape &shape_new, int32_t output_index);
 
   virtual Status LaunchTask(TaskContext &context) = 0;
@@ -58,6 +60,8 @@ class AicpuNodeTaskBase : public NodeTask {
   virtual Status InitForDependComputeTask() = 0;
 
   Status TaskCallback(TaskContext &context);
+  
+  virtual Status SetMemCopyTask(const domi::TaskDef &task_def) = 0;
 
   virtual Status UpdateIoAddr(TaskContext &context) = 0;
 
@@ -98,13 +102,13 @@ class AicpuTfNodeTask : public AicpuNodeTaskBase {
   Status TaskCallback(TaskContext &context) override;
 
   Status UpdateIoAddr(TaskContext &context) override;
-
- private:
-  Status SetMemCopyTask(const domi::TaskDef &task_def);
-
-  Status UpdateShapeAndDataByResultSummary(TaskContext &context);
+  
+  Status UpdateShapeAndDataByResultSummary(TaskContext &context) override;
   
   Status InitForDependComputeTask() override;
+
+  Status SetMemCopyTask(const domi::TaskDef &task_def) override;
+ private:
   ///
   /// read result summary and prepare copy task memory.
   /// @param context task context
@@ -166,7 +170,12 @@ class AicpuNodeTask : public AicpuNodeTaskBase {
   Status TaskCallback(TaskContext &context) override;
 
   Status UpdateIoAddr(TaskContext &context) override;
+  
+  Status UpdateShapeAndDataByResultSummary(TaskContext &context) override;
 
+  Status InitForDependComputeTask() override;
+
+  Status SetMemCopyTask(const domi::TaskDef &task_def);
  protected:
   // host mem
   std::unique_ptr<uint8_t[]> args_;