From 8e4328fd1999870cadfdcf652601c3430ff19bfd Mon Sep 17 00:00:00 2001 From: guopeian Date: Sat, 26 Jun 2021 14:30:24 +0800 Subject: [PATCH] fix --- ge/hybrid/node_executor/aicpu/aicpu_node_executor.h | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ge/hybrid/node_executor/aicpu/aicpu_node_executor.h b/ge/hybrid/node_executor/aicpu/aicpu_node_executor.h index d5fe2949..2bc90eaa 100644 --- a/ge/hybrid/node_executor/aicpu/aicpu_node_executor.h +++ b/ge/hybrid/node_executor/aicpu/aicpu_node_executor.h @@ -17,9 +17,9 @@ #ifndef GE_HYBRID_KERNEL_AICPU_NODE_EXECUTOR_H_ #define GE_HYBRID_KERNEL_AICPU_NODE_EXECUTOR_H_ +#include "hybrid/node_executor/node_executor.h" #include "external/graph/types.h" #include "cce/aicpu_engine_struct.h" -#include "hybrid/node_executor/node_executor.h" #include "aicpu_ext_info.h" namespace ge { @@ -51,7 +51,7 @@ class AicpuNodeTaskBase : public NodeTask { virtual Status UpdateOutputShapeFromExtInfo(TaskContext &task_context); - Status UpdateShapeAndDataByResultSummary(TaskContext &context); + virtual Status UpdateShapeAndDataByResultSummary(TaskContext &context); Status InitForDependComputeTask(bool is_tfkernel);