From 0dff7f1ac43bb7d074db36ce2ce44c52e6760e7e Mon Sep 17 00:00:00 2001
From: Cedric Nugteren <web@cedricnugteren.nl>
Date: Fri, 13 Apr 2018 22:09:16 +0200
Subject: Made GEMM rotation expectations kernel-specific

---
 src/clblast.cpp | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

(limited to 'src/clblast.cpp')

diff --git a/src/clblast.cpp b/src/clblast.cpp
index 331a39ef..10bb8cba 100644
--- a/src/clblast.cpp
+++ b/src/clblast.cpp
@@ -2490,7 +2490,8 @@ StatusCode GemmTempBufferSize(const Layout layout, const Transpose a_transpose,
     else {
       temp_buffer_size = Xgemm<T>::GetTempSize(layout, a_transpose, b_transpose, m, n, k,
                                                a_offset, a_ld, b_offset, b_ld, c_offset, c_ld,
-                                               db["MWG"], db["NWG"], db["KWG"]);
+                                               db["MWG"], db["NWG"], db["KWG"] * db["KREG"],
+                                               db["GEMMK"]);
     }
     temp_buffer_size *= sizeof(T); // translate from num-elements to bytes
     return StatusCode::kSuccess;
-- 
cgit v1.2.3