From 7c19a4c2fd03c8c67d35626ac821f6bdc4baa814 Mon Sep 17 00:00:00 2001 From: zjun Date: Mon, 13 Apr 2020 09:51:44 +0800 Subject: [PATCH] fix getnext op bug --- .../ccsrc/kernel/aicpu/aicpu_kernel_metadata.cc | 14 ++++++++------ 1 file changed, 8 insertions(+), 6 deletions(-) diff --git a/mindspore/ccsrc/kernel/aicpu/aicpu_kernel_metadata.cc b/mindspore/ccsrc/kernel/aicpu/aicpu_kernel_metadata.cc index 66750510698..9f91c1bdd2e 100644 --- a/mindspore/ccsrc/kernel/aicpu/aicpu_kernel_metadata.cc +++ b/mindspore/ccsrc/kernel/aicpu/aicpu_kernel_metadata.cc @@ -38,12 +38,14 @@ void AicpuMetadataInfo(const CNodePtr &kernel_node, std::vector inputs_format; - std::vector inputs_type; - for (size_t input_index = 0; input_index < AnfAlgo::GetInputTensorNum(kernel_node); ++input_index) { - inputs_format.emplace_back(kOpFormat_DEFAULT); - inputs_type.push_back(AnfAlgo::GetPrevNodeOutputInferDataType(kernel_node, input_index)); + if (op_name == kPrint || op_name == kGetNext) { + std::vector inputs_format{}; + std::vector inputs_type{}; + if (op_name == kPrint) { + for (size_t input_index = 0; input_index < AnfAlgo::GetInputTensorNum(kernel_node); ++input_index) { + inputs_format.emplace_back(kOpFormat_DEFAULT); + inputs_type.push_back(AnfAlgo::GetPrevNodeOutputInferDataType(kernel_node, input_index)); + } } std::vector outputs_format; std::vector outputs_type;