diff --git a/include/ck/tensor_operation/gpu/block/blockwise_softmax.hpp b/include/ck/tensor_operation/gpu/block/blockwise_softmax.hpp index d9f3bc7b43..7e62a822a8 100644 --- a/include/ck/tensor_operation/gpu/block/blockwise_softmax.hpp +++ b/include/ck/tensor_operation/gpu/block/blockwise_softmax.hpp @@ -15,12 +15,12 @@ namespace ck { /** * @brief Blockwise softmax * - * @tparam BlockSize - * @tparam AccDataType - * @tparam ThreadMap_M_K - * @tparam ThreadClusterDesc_M_K - * @tparam ThreadSliceDesc_M_K - * @tparam IgnoreNaN + * @tparam BlockSize Block size + * @tparam AccDataType Accumulator data type + * @tparam ThreadMap_M_K Thread id to m_k + * @tparam ThreadClusterDesc_M_K Threadwise cluster descriptor + * @tparam ThreadSliceDesc_M_K Threadwise slices descriptor + * @tparam IgnoreNaN Flag to ignore NaN, false by default */ template