From 162b3aefcf8d8c734e8956d2f16403a14ca7e52c Mon Sep 17 00:00:00 2001 From: jjfeing Date: Thu, 16 Jul 2020 15:43:15 +0800 Subject: [PATCH] add tbe build info with full name --- mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_build.cc | 2 ++ 1 file changed, 2 insertions(+) diff --git a/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_build.cc b/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_build.cc index 73642b291a4..7e920136572 100644 --- a/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_build.cc +++ b/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_build.cc @@ -43,6 +43,7 @@ constexpr auto kJInputs = "inputs"; constexpr auto kJOutputs = "outputs"; constexpr auto kJAttrs = "attrs"; constexpr auto kJKernelName = "kernel_name"; +constexpr auto kJFullName = "full_name"; constexpr auto kJOpInfo = "op_info"; constexpr auto kJDtype = "dtype"; constexpr auto kJtype = "type"; @@ -125,6 +126,7 @@ bool TbeKernelJsonCreator::GenTbeSingleKernelJson(const std::shared_ptrfullname_with_scope(); if (creater_type_ == SINGLE_BUILD) { TbeUtils::SaveJsonInfo(json_name_, json_info_); }