From 29c52b48a0f6a7b0f8fc1c6ede75768f06866669 Mon Sep 17 00:00:00 2001
From: Oleksandr Pavlyk <oleksandr.pavlyk@intel.com>
Date: Sun, 27 Aug 2023 11:58:50 -0500
Subject: [PATCH] Addressed coverity issue about handling bools in floor_divide

---
 .../elementwise_functions/floor_divide.hpp     | 18 ++++++++++++++++--
 1 file changed, 16 insertions(+), 2 deletions(-)

diff --git a/dpctl/tensor/libtensor/include/kernels/elementwise_functions/floor_divide.hpp b/dpctl/tensor/libtensor/include/kernels/elementwise_functions/floor_divide.hpp
index 268c679f00..32e97df58d 100644
--- a/dpctl/tensor/libtensor/include/kernels/elementwise_functions/floor_divide.hpp
+++ b/dpctl/tensor/libtensor/include/kernels/elementwise_functions/floor_divide.hpp
@@ -57,7 +57,12 @@ struct FloorDivideFunctor
 
     resT operator()(const argT1 &in1, const argT2 &in2)
     {
-        if constexpr (std::is_integral_v<argT1> || std::is_integral_v<argT2>) {
+        if constexpr (std::is_same_v<argT1, bool> &&
+                      std::is_same_v<argT2, bool>) {
+            return (in2) ? static_cast<resT>(in1) : resT(0);
+        }
+        else if constexpr (std::is_integral_v<argT1> ||
+                           std::is_integral_v<argT2>) {
             if (in2 == argT2(0)) {
                 return resT(0);
             }
@@ -81,7 +86,16 @@ struct FloorDivideFunctor
     sycl::vec<resT, vec_sz> operator()(const sycl::vec<argT1, vec_sz> &in1,
                                        const sycl::vec<argT2, vec_sz> &in2)
     {
-        if constexpr (std::is_integral_v<resT>) {
+        if constexpr (std::is_same_v<argT1, bool> &&
+                      std::is_same_v<argT2, bool>) {
+            sycl::vec<resT, vec_sz> res;
+#pragma unroll
+            for (int i = 0; i < vec_sz; ++i) {
+                res[i] = (in2[i]) ? static_cast<resT>(in1[i]) : resT(0);
+            }
+            return res;
+        }
+        else if constexpr (std::is_integral_v<resT>) {
             sycl::vec<resT, vec_sz> res;
 #pragma unroll
             for (int i = 0; i < vec_sz; ++i) {