Skip to content

GitLab

Unverified Commit cff7fab7 authored by Bartłomiej Kocot's avatar Bartłomiej Kocot Committed by GitHub
Browse files

[CK TILE] Fix universal gemm template keywords (#1704)

parent 28e02cf5
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
include/ck_tile/ops/gemm/block/block_universal_gemm_as_bs_cr.hpp
View file @ cff7fab7
...@@ -623,7 +623,7 @@ struct BlockUniversalGemmAsBsCr ...@@ -623,7 +623,7 @@ struct BlockUniversalGemmAsBsCr
CK_TILE_DEVICE void LocalPrefetch(const ASmemBlockWindow& a_block_window, CK_TILE_DEVICE void LocalPrefetch(const ASmemBlockWindow& a_block_window,
const BSmemBlockWindow& b_block_window) const BSmemBlockWindow& b_block_window)
{ {
block_gemm_impl_.template LocalPrefetch(a_block_window, b_block_window); block_gemm_impl_.LocalPrefetch(a_block_window, b_block_window);
} }
// C += A * B // C += A * B
...@@ -632,7 +632,7 @@ struct BlockUniversalGemmAsBsCr ...@@ -632,7 +632,7 @@ struct BlockUniversalGemmAsBsCr
const ASmemBlockWindow& a_block_window, const ASmemBlockWindow& a_block_window,
const BSmemBlockWindow& b_block_window) const BSmemBlockWindow& b_block_window)
{ {
block_gemm_impl_.template operator()(c_block_tensor, a_block_window, b_block_window); block_gemm_impl_(c_block_tensor, a_block_window, b_block_window);
} }
// C = A * B // C = A * B
...@@ -641,7 +641,7 @@ struct BlockUniversalGemmAsBsCr ...@@ -641,7 +641,7 @@ struct BlockUniversalGemmAsBsCr
const BSmemBlockWindow& b_block_window) const BSmemBlockWindow& b_block_window)
{ {
auto c_block_tensor = MakeCBlockTile(); auto c_block_tensor = MakeCBlockTile();
block_gemm_impl_.template operator()(c_block_tensor, a_block_window, b_block_window); block_gemm_impl_(c_block_tensor, a_block_window, b_block_window);
return c_block_tensor; return c_block_tensor;
} }
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment