Fixed half-precision bugs in HTBMV/HTPMV/HTRMV/HSYR2K/HTRMM related to incorrect constants

2 files changed, 3 insertions, 3 deletions

diff --git a/src/routines/level3/xsyr2k.cpp b/src/routines/level3/xsyr2k.cpp
index fdef43dc..7900eb74 100644
--- a/src/routines/level3/xsyr2k.cpp
+++ b/src/routines/level3/xsyr2k.cpp
@@ -149,7 +149,7 @@ void Xsyr2k<T>::DoSyr2k(const Layout layout, const Triangle triangle, const Tran
   eventWaitList.push_back(eventKernel1);
 
   // Swaps the arguments for matrices A and B, and sets 'beta' to 1
-  auto one = static_cast<T>(1);
+  auto one = ConstantOne<T>();
   kernel.SetArgument(3, GetRealArg(one));
   kernel.SetArgument(4, b_temp());
   kernel.SetArgument(5, a_temp());
diff --git a/src/routines/level3/xtrmm.cpp b/src/routines/level3/xtrmm.cpp
index 02c295ac..26ef2a5e 100644
--- a/src/routines/level3/xtrmm.cpp
+++ b/src/routines/level3/xtrmm.cpp
@@ -101,7 +101,7 @@ void Xtrmm<T>::DoTrmm(const Layout layout, const Side side, const Triangle trian
            alpha,
            temp_triangular, 0, k,
            b_buffer_copy, b_offset, b_ld,
-           static_cast<T>(0.0),
+           ConstantZero<T>(),
            b_buffer, b_offset, b_ld);
   }
 
@@ -113,7 +113,7 @@ void Xtrmm<T>::DoTrmm(const Layout layout, const Side side, const Triangle trian
              alpha,
              b_buffer_copy, b_offset, b_ld,
              temp_triangular, 0, k,
-             static_cast<T>(0.0),
+             ConstantZero<T>(),
              b_buffer, b_offset, b_ld);
     } catch (BLASError &e) {
       // A and B are now reversed, so also reverse the error codes returned from the Xgemm routine