|
|
|
@@ -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; |
|
|
|
|