delegate bhalf2 case to bhalf inner product

a0d36aea · Max Podkorytov · d6557188 · a0d36aea
Commit a0d36aea authored Nov 17, 2023 by Max Podkorytov
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 2 deletions

include/ck/utility/inner_product.hpp include/ck/utility/inner_product.hpp +3 -2

No files found.
--- a/include/ck/utility/inner_product.hpp
+++ b/include/ck/utility/inner_product.hpp
@@ -91,8 +91,9 @@ __device__ void inner_product<bhalf2_t, bhalf2_t, float>(const bhalf2_t& a, cons
    const vector_type<bhalf_t, 2> b_vector{b};

    static_for<0, 2, 1>{}([&](auto i) {
-        c += type_convert<float>(a_vector.AsType<bhalf_t>()[i]) *
-             type_convert<float>(b_vector.AsType<bhalf_t>()[i]);
+        inner_product(a_vector.AsType<bhalf_t>()[i], 
+                      b_vector.AsType<bhalf_t>()[i],
+                      c);
    });
 }