Added bf16 to condition

nikita-savelyevv · Nov 21, 2023 · 5086e18 · 5086e18
1 parent d1f71ee
commit 5086e18
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/common/transformations/src/transformations/convert_precision.cpp b/src/common/transformations/src/transformations/convert_precision.cpp
@@ -367,7 +367,7 @@ bool ov::pass::ConvertPrecision::run_on_model(const std::shared_ptr<ov::Model>&
     if (used_precisions.empty())
         return false;
 
-    bool has_fp16_compression = m_precisions.count(element::f32) > 0 && m_precisions[element::f32] == element::f16;
+    bool has_fp16_compression = m_precisions.count(element::f32) > 0 && (m_precisions[element::f32] == element::f16 || m_precisions[element::f32] == element::bf16);
 
     if (m_keep_precision_sensitive_in_fp32 && has_fp16_compression) {
         pass::Manager manager(get_pass_config());