Made GEMM rotation expectations kernel-specific

1 files changed, 2 insertions, 1 deletions

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;