Browse Source

fix lhisi cast be deleted question when fp32 input

tags/v1.3.0
wxl 4 years ago
parent
commit
b29adb30e3
1 changed files with 3 additions and 0 deletions
  1. +3
    -0
      ge/graph/passes/cast_remove_pass.cc

+ 3
- 0
ge/graph/passes/cast_remove_pass.cc View File

@@ -123,6 +123,7 @@ Status CastRemovePass::DoFuse(const OpsKernelManager &ops_kernel_manager,
in_desc->SetDataType(type);
out_desc->SetDataType(type);
bool is_supported = false;
string un_supported_reasons;
for (const auto &ops_kernel_store_info : ops_kernel_manager.GetAllOpsKernelInfoStores()) {
map<string, OpInfo> op_infos;
ops_kernel_store_info.second->GetAllOpsKernelInfo(op_infos);
@@ -134,12 +135,14 @@ Status CastRemovePass::DoFuse(const OpsKernelManager &ops_kernel_manager,
if (is_supported) {
break;
}
un_supported_reasons += "{op_store " + ops_kernel_store_info.first + ":" + un_supported_reason + "} ";
}
if (!is_supported) {
// if no operator_info_store supported, do nothing
in_desc->SetDataType(in_desc_org_dtype);
out_desc->SetDataType(out_desc_org_dtype);
to_be_deleted_cast_index.clear();
GELOGI("Fused Op[%s] check supported fail! Reasons is as follows: %s", un_supported_reasons.c_str());
return SUCCESS;
}



Loading…
Cancel
Save