|
|
@@ -1707,7 +1707,7 @@ Status ModelManager::CheckAicpuOpList(GeModelPtr ge_model) { |
|
|
bool aicpu_need_check = ge::AttrUtils::GetListStr(ge_model, "needCheckCpu", aicpu_optype_list); |
|
|
bool aicpu_need_check = ge::AttrUtils::GetListStr(ge_model, "needCheckCpu", aicpu_optype_list); |
|
|
bool tf_need_check = ge::AttrUtils::GetListStr(ge_model, "needCheckTf", aicpu_tf_optype_list); |
|
|
bool tf_need_check = ge::AttrUtils::GetListStr(ge_model, "needCheckTf", aicpu_tf_optype_list); |
|
|
if (!aicpu_need_check && !tf_need_check) { |
|
|
if (!aicpu_need_check && !tf_need_check) { |
|
|
GELOGI("No need to check aicpu optype."); |
|
|
|
|
|
|
|
|
GELOGI("Graph:%s No need to check aicpu optype.", ge_model->GetGraph().GetName().c_str()); |
|
|
return SUCCESS; |
|
|
return SUCCESS; |
|
|
} |
|
|
} |
|
|
GE_CHK_STATUS_RET(LaunchKernelCheckAicpuOp(aicpu_optype_list, aicpu_tf_optype_list), "Launch check aicpu op type failed."); |
|
|
GE_CHK_STATUS_RET(LaunchKernelCheckAicpuOp(aicpu_optype_list, aicpu_tf_optype_list), "Launch check aicpu op type failed."); |
|
|
|