p47068915
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix kernel_build_info is nullptr

This commit is contained in:
lby 2022-08-15 17:41:27 +08:00
parent 984a92c0b8
commit a0fcfb08f5
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mindspore/ccsrc/common/graph_kernel/adapter/callback_impl.cc
View File

@ -191,10 +191,14 @@ void CallbackImpl::SetEmptyKernelInfo(const AnfNodePtr &node) {
} }
void CallbackImpl::ResetKernelInfo(const AnfNodePtr &node) { void CallbackImpl::ResetKernelInfo(const AnfNodePtr &node) {
MS_EXCEPTION_IF_NULL(node);
auto cnode = node->cast<CNodePtr>(); auto cnode = node->cast<CNodePtr>();
MS_EXCEPTION_IF_NULL(cnode); MS_EXCEPTION_IF_NULL(cnode);
if (GetTargetFromContext() == kAscendDevice) { if (GetTargetFromContext() == kAscendDevice) {
auto kernel_info = cnode->kernel_info_ptr();
if (kernel_info == nullptr) {
cnode->set_kernel_info(std::make_shared<device::KernelInfo>()); cnode->set_kernel_info(std::make_shared<device::KernelInfo>());
}
auto kernel_info_setter = GraphKernelInfoManager::Instance().GetGraphKernelInfo(kAscendDevice); auto kernel_info_setter = GraphKernelInfoManager::Instance().GetGraphKernelInfo(kAscendDevice);
MS_EXCEPTION_IF_NULL(kernel_info_setter); MS_EXCEPTION_IF_NULL(kernel_info_setter);
kernel_info_setter->SetKernelInfo(cnode, KernelType::UNKNOWN_KERNEL_TYPE); kernel_info_setter->SetKernelInfo(cnode, KernelType::UNKNOWN_KERNEL_TYPE);