limit quant with external extension

This commit is contained in:
yeyunpeng2020 2021-10-26 12:45:02 +08:00
parent f86795c16c
commit 6d133eaa61
1 changed files with 4 additions and 0 deletions

View File

@ -515,6 +515,10 @@ FuncGraphPtr AnfTransform::TransformFuncGraph(const FuncGraphPtr &old_graph, con
return nullptr;
}
if (!config->pluginsPath.empty() && config->commonQuantParam.quant_type != schema::QuantType_QUANT_NONE) {
MS_LOG(ERROR) << "Unsupported external extension with quantization.";
return nullptr;
}
status = DoQuantize(old_graph, config);
if (status != RET_OK) {
MS_LOG(ERROR) << "Do Quantize failed.";