From cff37a65a3834d29a2b1803ed447fd3dae249155 Mon Sep 17 00:00:00 2001
From: guopeian <guopeian1@hisilicon.com>
Date: Tue, 20 Jul 2021 14:02:41 +0800
Subject: [PATCH] fix

---
 ge/single_op/task/op_task.h | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/ge/single_op/task/op_task.h b/ge/single_op/task/op_task.h
index 1769d976..a0d5b574 100644
--- a/ge/single_op/task/op_task.h
+++ b/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 {