check min<=max in ChooseQuantizationParams

Summary: pytorch#1590 Differential Revision: D43383474 fbshipit-source-id: ff27b44b589f8dbde54f3a2c830d7d489f7a5138
jspark1105 · Feb 24, 2023 · 9ee9edc · 9ee9edc
1 parent 7f791ed
commit 9ee9edc
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/QuantUtils.cc b/src/QuantUtils.cc
@@ -36,6 +36,11 @@ TensorQuantizationParams ChooseQuantizationParams(
     int32_t qmax,
     bool preserve_sparsity,
     bool force_scale_power_of_two) {
+  if (min > max) {
+    throw std::runtime_error(
+        "In ChooseQuantizationParams, min should be less than or equal to max");
+  }
+
   if (min < 0 && max > 0 && preserve_sparsity) {
     int symmetric_qmin = -((qmax - qmin) / 2 + 1);
     int symmetric_qmax = (qmax - qmin) / 2;