From fad43b902dd7923d2cb7d2ec45d7c36a13226419 Mon Sep 17 00:00:00 2001 From: limingqi107 Date: Sun, 28 Jun 2020 21:10:58 +0800 Subject: [PATCH] fix cast kernel build in pynative mode --- mindspore/ccsrc/kernel/akg/akg_kernel_attrs_process.cc | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/mindspore/ccsrc/kernel/akg/akg_kernel_attrs_process.cc b/mindspore/ccsrc/kernel/akg/akg_kernel_attrs_process.cc index 3a0cc3eb25..018fbe4f2a 100644 --- a/mindspore/ccsrc/kernel/akg/akg_kernel_attrs_process.cc +++ b/mindspore/ccsrc/kernel/akg/akg_kernel_attrs_process.cc @@ -68,10 +68,10 @@ void SetAkgAttrsForFive2Four(const AnfNodePtr &anf_node) { void SetAkgAttrsForCast(const AnfNodePtr &anf_node) { MS_EXCEPTION_IF_NULL(anf_node); // The x and output are akg op input and output param. - std::vector input_names = {"x"}; + std::vector input_names = {"x", "dst_type"}; std::vector output_names = {"output"}; - AnfAlgo::SetNodeAttr("input_names", MakeValue(input_names), anf_node); - AnfAlgo::SetNodeAttr("output_names", MakeValue(output_names), anf_node); + AnfAlgo::SetNodeAttr(kAttrInputNames, MakeValue(input_names), anf_node); + AnfAlgo::SetNodeAttr(kAttrOutputNames, MakeValue(output_names), anf_node); std::string dst_type; TypeId output_type = AnfAlgo::GetOutputDeviceDataType(anf_node, 0);