Fix round() output dtype for boolean input to match numpy

vlad-perevezentsev · vlad-perevezentsev · commit 6f5a792c8542 · 2026-03-13T03:55:55.000-07:00
diff --git a/dpctl_ext/tensor/libtensor/include/kernels/elementwise_functions/round.hpp b/dpctl_ext/tensor/libtensor/include/kernels/elementwise_functions/round.hpp
@@ -116,6 +116,7 @@ template <typename T>
 struct RoundOutputType
 {
     using value_type = typename std::disjunction<
+        td_ns::TypeMapResultEntry<T, bool, sycl::half>,
         td_ns::TypeMapResultEntry<T, std::uint8_t>,
         td_ns::TypeMapResultEntry<T, std::uint16_t>,
         td_ns::TypeMapResultEntry<T, std::uint32_t>,

Original file line number	Diff line number	Diff line change
`@@ -116,6 +116,7 @@ template <typename T>`
`116`	`116`	`struct RoundOutputType`
`117`	`117`	`{`
`118`	`118`	`using value_type = typename std::disjunction<`
	`119`	`+ td_ns::TypeMapResultEntry<T, bool, sycl::half>,`
`119`	`120`	`td_ns::TypeMapResultEntry<T, std::uint8_t>,`
`120`	`121`	`td_ns::TypeMapResultEntry<T, std::uint16_t>,`
`121`	`122`	`td_ns::TypeMapResultEntry<T, std::uint32_t>,`