[dalgona] Add bool type (#14248)

This commit adds bool type in numpyArray function. ONE-DCO-1.0-Signed-off-by: seongwoo <[email protected]>
Samsung · Oct 24, 2024 · 9d5e68d · 9d5e68d
1 parent a1612da
commit 9d5e68d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/compiler/dalgona/src/Utils.cpp b/compiler/dalgona/src/Utils.cpp
@@ -68,6 +68,8 @@ py::array numpyArray(const Tensor *tensor)
       return py::array_t<int64_t, py::array::c_style>(shape, tensor->data<int64_t>());
     case loco::DataType::U8:
       return py::array_t<uint8_t, py::array::c_style>(shape, tensor->data<uint8_t>());
+    case loco::DataType::BOOL:
+      return py::array_t<bool, py::array::c_style>(shape, tensor->data<bool>());
     default:
       throw std::runtime_error("Unsupported data type");
   }