| @@ -21,6 +21,7 @@ | |||||
| #include "cce/aicpu_engine_struct.h" | #include "cce/aicpu_engine_struct.h" | ||||
| #include "hybrid/node_executor/node_executor.h" | #include "hybrid/node_executor/node_executor.h" | ||||
| #include "aicpu_ext_info.h" | #include "aicpu_ext_info.h" | ||||
| #include "common/ge_inner_error_codes.h" | |||||
| namespace ge { | namespace ge { | ||||
| namespace hybrid { | namespace hybrid { | ||||
| @@ -41,7 +42,9 @@ class AicpuNodeTaskBase : public NodeTask { | |||||
| virtual Status Init(const HybridModel &model) = 0; | virtual Status Init(const HybridModel &model) = 0; | ||||
| virtual Status SetMemCopyTask(const domi::TaskDef &task_def) = 0; | |||||
| virtual Status SetMemCopyTask(const domi::TaskDef &task_def) { | |||||
| return UNSUPPORTED; | |||||
| } | |||||
| Status UpdateArgs(TaskContext &context) override; | Status UpdateArgs(TaskContext &context) override; | ||||
| @@ -161,10 +164,6 @@ class AicpuNodeTask : public AicpuNodeTaskBase { | |||||
| Status Init(const HybridModel &model) override; | Status Init(const HybridModel &model) override; | ||||
| Status SetMemCopyTask(const domi::TaskDef &task_def) override { | |||||
| return UNSUPPORTED; | |||||
| } | |||||
| protected: | protected: | ||||
| Status LaunchTask(TaskContext &context) override; | Status LaunchTask(TaskContext &context) override; | ||||