mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-05 14:11:29 +00:00
Documentation for newly added struct (#2051)
This commit is contained in:
@@ -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,
|
||||
|
||||
Reference in New Issue
Block a user