From 00281dad26c6e83ef39629cc26b24154f3219e06 Mon Sep 17 00:00:00 2001 From: Cedric Nugteren Date: Mon, 27 Feb 2017 21:00:04 +0100 Subject: Fixed half-precision bugs in HTBMV/HTPMV/HTRMV/HSYR2K/HTRMM related to incorrect constants --- src/routines/level3/xsyr2k.cpp | 2 +- src/routines/level3/xtrmm.cpp | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) (limited to 'src/routines/level3') 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::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(1); + auto one = ConstantOne(); 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::DoTrmm(const Layout layout, const Side side, const Triangle trian alpha, temp_triangular, 0, k, b_buffer_copy, b_offset, b_ld, - static_cast(0.0), + ConstantZero(), b_buffer, b_offset, b_ld); } @@ -113,7 +113,7 @@ void Xtrmm::DoTrmm(const Layout layout, const Side side, const Triangle trian alpha, b_buffer_copy, b_offset, b_ld, temp_triangular, 0, k, - static_cast(0.0), + ConstantZero(), 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 -- cgit v1.2.3