Commit 8d988972 authored by j4yan's avatar j4yan
Browse files

formatting

parent 9c2820ef
...@@ -20,12 +20,14 @@ namespace device { ...@@ -20,12 +20,14 @@ namespace device {
* \brief Wrapper function of GridwiseGemm::Run to realize BatchedGEMM. * \brief Wrapper function of GridwiseGemm::Run to realize BatchedGEMM.
* *
* \tparam ComputePrtOffsetOfBatch Class that computes the base pointer offsets of A, B, C matrix * \tparam ComputePrtOffsetOfBatch Class that computes the base pointer offsets of A, B, C matrix
* given the batch. For example, ComputePtrOffsetOfStridedBatch() computes the offsets of evenly strided batched, but we can easily extend to other layouts. The returned offset can be either \p index_t or \p long_index_t. If it returns * given the batch. For example, ComputePtrOffsetOfStridedBatch() computes the offsets of evenly
* \p long_index_t, we are not subject to the 2GB limitations. * strided batched, but we can easily extend to other layouts. The returned offset can be either \p
* index_t or \p long_index_t. If it returns \p long_index_t, we are not subject to the 2GB
* limitations.
* *
* \tparam * \tparam Block2CTileMap Block2CTileMap::CalculateBottomIndex() takes in id of a workgroup and
* Block2CTileMap Block2CTileMap::CalculateBottomIndex() takes in id of a workgroup and returns the * returns the 2D index of the tile that it computes. \see
* 2D index of the tile that it computes. \see GridwiseGemm_k0mk1_k0nk1_mn_xdlops_v2r3::Run(). * GridwiseGemm_k0mk1_k0nk1_mn_xdlops_v2r3::Run().
* *
* \note Using \p ComputePrtOffsetOfBatch gives us the flexibility that 2 workgroups can compute 2 * \note Using \p ComputePrtOffsetOfBatch gives us the flexibility that 2 workgroups can compute 2
* tiles from different matrices. Keep in mind that these 2 matrices can share the same grid * tiles from different matrices. Keep in mind that these 2 matrices can share the same grid
......
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