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

GetAttr nullptr check

This commit is contained in:
hangangqiang 2021-09-29 10:38:41 +08:00
parent 338ab157c7
commit a8ff75b0e8
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mindspore/lite/tools/optimizer/fisson/fisson_util.cc
View File

@ -187,7 +187,9 @@ std::shared_ptr<ops::Conv2DFusion> CopyConvPrim(const std::shared_ptr<ops::Conv2
new_prim->set_activation_type(ori_conv_prim->get_activation_type());
new_prim->set_pad_list(ori_conv_prim->get_pad_list());
if (ori_conv_prim->GetAttr(ops::kIsDepthWise) != nullptr) {
bool is_depth_wise = GetValue<bool>(ori_conv_prim->GetAttr(ops::kIsDepthWise));
auto is_depth_value = ori_conv_prim->GetAttr(ops::kIsDepthWise);
MS_CHECK_TRUE_MSG(is_depth_value != nullptr, nullptr, "conv has no kIsDepthWise attribute");
bool is_depth_wise = GetValue<bool>(is_depth_value);
new_prim->AddAttr(ops::kIsDepthWise, MakeValue<bool>(is_depth_wise));
}
return new_prim;