Browse Source

Remove gentask in DEPEND_COMPUTE task executor.

pull/913/head
unknown 5 years ago
parent
commit
e105bfcb00
1 changed files with 4 additions and 5 deletions
  1. +4
    -5
      ge/hybrid/node_executor/aicpu/aicpu_node_executor.h

+ 4
- 5
ge/hybrid/node_executor/aicpu/aicpu_node_executor.h View File

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


Loading…
Cancel
Save