Merge pull request #381 from kokkos/issue-368

Fix GEMM scratch memory size
kokkos · Jan 30, 2019 · 67af7da · 67af7da
2 parents 52a6ba2 + 1767da4
commit 67af7da
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/blas/impl/KokkosBlas3_gemm_impl.hpp b/src/blas/impl/KokkosBlas3_gemm_impl.hpp
@@ -510,9 +510,9 @@ struct GEMMImpl {
   void run(int team_size, int vector_length, int scr_level) {
     scratch_level = scr_level;
     int scratch_memory_size =
-      ViewTypeAScratch::required_allocation_size() +
-      ViewTypeBScratch::required_allocation_size() +
-      ViewTypeCScratch::required_allocation_size();
+      ViewTypeAScratch::shmem_size() +
+      ViewTypeBScratch::shmem_size() +
+      ViewTypeCScratch::shmem_size();
 
     Kokkos::TeamPolicy<ExecSpace,Kokkos::LaunchBounds<384,2>> policy(num_blocks_0*num_blocks_1,team_size,vector_length);