average modified to fit cayley-dickson

jfalcou · Aug 27, 2023 · 40be733 · 40be733
1 parent eac3271
commit 40be733
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/include/eve/module/core/regular/impl/average.hpp b/include/eve/module/core/regular/impl/average.hpp
@@ -41,8 +41,14 @@ average_(EVE_SUPPORTS(cpu_), T const& a, T const& b) noexcept
 {
   if constexpr(has_native_abi_v<T>)
   {
-    if constexpr( integral_value<T> ) return (a & b) + ((a ^ b) >> 1); //compute ceil( (x+y)/2 )
-    else return fma(a, half(eve::as(a)), b * half(eve::as(a)));
+    if constexpr( integral_value<T> )
+      return (a & b) + ((a ^ b) >> 1); //compute ceil( (x+y)/2 )
+    else
+    {
+      using u_t =  underlying_type_t<T>;
+      const auto h = eve::half(eve::as<u_t>());
+      return fma(a, h, b * h);
+    }
   }
   else return apply_over(average, a, b);
 }