From 11bf5400188c0801b52abb16e0a64ae5c6c93794 Mon Sep 17 00:00:00 2001
From: Maksim Derbasov <maksim.derbasov@intel.com>
Date: Mon, 21 Feb 2022 14:24:27 +0300
Subject: [PATCH] Simple patch for fix random bool vector generation (#10493)

* Dirty patch for fix bool generation

* Bernoulli distribution for bool
---
 .../include/ngraph_functions/utils/data_utils.hpp           | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/src/tests/ngraph_helpers/ngraph_functions/include/ngraph_functions/utils/data_utils.hpp b/src/tests/ngraph_helpers/ngraph_functions/include/ngraph_functions/utils/data_utils.hpp
index 8e7dc75d6ad..ee7d6ed7184 100644
--- a/src/tests/ngraph_helpers/ngraph_functions/include/ngraph_functions/utils/data_utils.hpp
+++ b/src/tests/ngraph_helpers/ngraph_functions/include/ngraph_functions/utils/data_utils.hpp
@@ -35,6 +35,12 @@ generateVector(size_t vec_len,
             res[i] = static_cast<dataType>(dist(gen));
         }
         return res;
+    } else if (std::is_same<bool, dataType>()) {
+        std::bernoulli_distribution dist;
+        for (size_t i = 0; i < vec_len; i++) {
+            res[i] = static_cast<dataType>(dist(gen));
+        }
+        return res;
     } else {
         // chose values between this range to avoid type overrun (e.g. in case of I8 precision)
         std::uniform_int_distribution<long> dist(static_cast<long>(startFrom), static_cast<long>(upTo));