fix

ge/single_op/task/op_task.h

@@ -262,7 +262,7 @@ class AiCpuCCTask : public AiCpuBaseTask {
   ~AiCpuCCTask() override;
   AiCpuCCTask(const AiCpuCCTask &) = delete;
   AiCpuCCTask &operator=(const AiCpuCCTask &) = delete;
-
+  Status SetMemCopyTask(const domi::KernelDef &kernel_def);
   Status LaunchKernel(rtStream_t stream) override;
   void GetIoAddr(uintptr_t *&arg_base, size_t &arg_count) override;
   const void *GetArgs() const;
@@ -271,7 +271,10 @@ class AiCpuCCTask : public AiCpuBaseTask {
   void SetkernelName(const std::string &kernel_Name);
   void SetIoAddr(uintptr_t *io_addr);
   size_t GetArgSize() const;
+ private:
+  Status InitForSummaryAndCopy();
 
+  Status CopyDataToHbm(vector<DataBuffer> &outputs, rtStream_t stream) override;
 private:
   friend class AiCpuCCTaskBuilder;
   std::string so_name_;
@@ -285,6 +288,13 @@ private:
   uint32_t dump_flag_ = RT_KERNEL_DEFAULT;
   std::string op_type_;
   uint64_t kernel_id_ = 0;
+  // host memcpy mem
+  std::unique_ptr<uint8_t[]> memcpy_args_;
+  std::string memcpy_so_name_;
+  std::string memcpy_kernel_name_;
+  std::vector<uint64_t> copy_io_addr_;
+  // args size
+  uint32_t memcpy_args_size_ = 0;
 };
 
 class MemcpyAsyncTask : public OpTask {