build fix

openvinotoolkit · Aug 6, 2024 · 28a4669 · 28a4669
1 parent b6a15c6
commit 28a4669
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/common/transformations/src/ov_ops/dynamic_quantize.cpp b/src/common/transformations/src/ov_ops/dynamic_quantize.cpp
@@ -57,8 +57,10 @@ std::vector<ov::PartialShape> DynamicQuantize::shape_infer(const DynamicQuantize
         if (group_sizes[i] == UINT64_MAX)
             scale_shape[i] = 1;
         else {
-            scale_shape[i] /= group_sizes[i];  // if group_size is larger than shape, scale_shape will be 1
-            scale_shape[i] = std::max(scale_shape[i].get_length(), 1L);
+            // if group_size is larger than shape, scale_shape will be 1
+            scale_shape[i] /= group_sizes[i];
+            if (scale_shape[i] == 0)
+                scale_shape[i] = 1;
         }
     }
     out_shapes.push_back(scale_shape);