Minor fix to transpose local memory size

codeplaysoftware · Jul 20, 2023 · 18a2701 · 18a2701
1 parent 4ed8a3d
commit 18a2701
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/interface/extension_interface.hpp b/src/interface/extension_interface.hpp
@@ -79,8 +79,8 @@ typename sb_handle_t::event_t _transpose_outplace_impl(
           in_view, _inc_in, out_view, _inc_out, _alpha);
 
   if constexpr (local_memory) {
-    index_t local_mem = static_cast<index_t>(
-        (num_line_elems + 1) * num_line_elems / num_tiles_per_line);
+    index_t local_mem = static_cast<index_t>((num_line_elems + 1) * Tile_size /
+                                             num_tiles_per_line);
     return sb_handle.execute(trans_scale_tree, static_cast<index_t>(wg_size),
                              global_size, local_mem);
   } else {