From b13a4746cbf47f92eb40934134e220d4ec8b4a81 Mon Sep 17 00:00:00 2001 From: guopeian Date: Thu, 24 Jun 2021 21:46:47 +0800 Subject: [PATCH] fix ut --- ge/hybrid/node_executor/aicpu/aicpu_node_executor.h | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) diff --git a/ge/hybrid/node_executor/aicpu/aicpu_node_executor.h b/ge/hybrid/node_executor/aicpu/aicpu_node_executor.h index 02180b1d..8980d41b 100644 --- a/ge/hybrid/node_executor/aicpu/aicpu_node_executor.h +++ b/ge/hybrid/node_executor/aicpu/aicpu_node_executor.h @@ -55,12 +55,11 @@ class AicpuNodeTaskBase : public NodeTask { virtual Status LaunchTask(TaskContext &context) = 0; - virtual Status TaskCallback(TaskContext &context) = 0; - virtual Status UpdateIoAddr(TaskContext &context) = 0; static Status AllocTensorBuffer(size_t size, std::unique_ptr &tensor_buffer); - + private: + Status TaskCallback(TaskContext &context); protected: const NodeItem *node_item_; // just reference. @@ -93,8 +92,6 @@ class AicpuTfNodeTask : public AicpuNodeTaskBase { Status LaunchTask(TaskContext &context) override; - Status TaskCallback(TaskContext &context) override; - Status UpdateIoAddr(TaskContext &context) override; private: @@ -162,7 +159,7 @@ class AicpuNodeTask : public AicpuNodeTaskBase { Status LaunchTask(TaskContext &context) override; - Status TaskCallback(TaskContext &context) override; + override; Status UpdateIoAddr(TaskContext &context) override; @@ -171,6 +168,8 @@ class AicpuNodeTask : public AicpuNodeTaskBase { Status InitForDependComputeTask(); + Status UpdateShapeAndDataByResultSummary(TaskContext &context); + protected: // host mem std::unique_ptr args_;