diff --git a/ge/hybrid/node_executor/aicpu/aicpu_node_executor.h b/ge/hybrid/node_executor/aicpu/aicpu_node_executor.h index 401d22a6..bbe60ee8 100644 --- a/ge/hybrid/node_executor/aicpu/aicpu_node_executor.h +++ b/ge/hybrid/node_executor/aicpu/aicpu_node_executor.h @@ -21,6 +21,7 @@ #include "cce/aicpu_engine_struct.h" #include "hybrid/node_executor/node_executor.h" #include "aicpu_ext_info.h" +#include "common/ge_inner_error_codes.h" namespace ge { namespace hybrid { @@ -41,7 +42,9 @@ class AicpuNodeTaskBase : public NodeTask { 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; @@ -161,10 +164,6 @@ class AicpuNodeTask : public AicpuNodeTaskBase { Status Init(const HybridModel &model) override; - Status SetMemCopyTask(const domi::TaskDef &task_def) override { - return UNSUPPORTED; - } - protected: Status LaunchTask(TaskContext &context) override;