Add corner case in scale calculation (#37352)

4d891c00 · joanna.wozna.intel · GitHub · bb2733fa · 4d891c00
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

paddle/fluid/inference/api/mkldnn_quantizer.cc paddle/fluid/inference/api/mkldnn_quantizer.cc +1 -1

未找到文件。
--- a/paddle/fluid/inference/api/mkldnn_quantizer.cc
+++ b/paddle/fluid/inference/api/mkldnn_quantizer.cc
@@ -306,7 +306,7 @@ AnalysisPredictor::MkldnnQuantizer::GetKLScalingFactor(
    std::vector<int> reference_distr_P(&hist[0], &hist[i]);
    auto outliers_count =
        std::accumulate(&hist[i], &hist[precision_hist_num_bins], 0);
-    if (reference_distr_P[i - 1] == 0) {
+    if (i <= 0 || reference_distr_P[i - 1] == 0) {
      continue;
    }
    reference_distr_P[i - 1] += outliers_count;