Documentation for newly added struct (#2051)

This commit is contained in:
Khushbu Agarwal
2025-04-03 16:24:34 -07:00
committed by GitHub
parent 572cd820ce
commit b443056a26
2 changed files with 17 additions and 3 deletions

View File

@@ -54,6 +54,17 @@ struct DataTypeTraits<ck_tile::pk_int4_t>
static constexpr const char* name = "pk_int4_t";
};
/**
* @brief trait for GEMM kernel
* @param pipeline: pipeline name
* @param scheduler: scheduler name
* @param epilogue: epilogue name
* @param kPadM: padding for M dimension
* @param kPadN: padding for N dimension
* @param kPadK: padding for K dimension
*
*/
struct KernelTraits
{
std::string pipeline;
@@ -173,7 +184,11 @@ void permute_vectors_i4x4_b(Tensor& tensor)
}
}
// verification code
/**
* @brief Function to verify the kernel output with reference implementation on CPU/GPU
*
*/
template <typename ADataType,
typename BDataType,
typename AccDataType,