Skip to content

Commit

Permalink
[CK TILE] Fix universal gemm template keywords (#1704)
Browse files Browse the repository at this point in the history
  • Loading branch information
bartekxk authored Nov 30, 2024
1 parent 28e02cf commit cff7fab
Showing 1 changed file with 3 additions and 3 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -623,7 +623,7 @@ struct BlockUniversalGemmAsBsCr
CK_TILE_DEVICE void LocalPrefetch(const ASmemBlockWindow& a_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
Expand All @@ -632,7 +632,7 @@ struct BlockUniversalGemmAsBsCr
const ASmemBlockWindow& a_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
Expand All @@ -641,7 +641,7 @@ struct BlockUniversalGemmAsBsCr
const BSmemBlockWindow& b_block_window)
{
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;
}

Expand Down

0 comments on commit cff7fab

Please sign in to comment.