mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-06-30 03:37:38 +00:00
Fix example compile error
This commit is contained in:
@@ -500,8 +500,8 @@ bool run_mx_gemm(const ProblemSizeSplitK& problem_size, const ExecutionConfig& c
|
||||
// partial sums(K/ScaleBlockSize)]
|
||||
// FLOPS = 2 * M * N * K + 2 * M * N * K / ScaleBlockSize
|
||||
std::size_t flop = std::size_t(2) * M * N * K + std::size_t(2) * M * N * K / ScaleBlockSize;
|
||||
std::size_t num_btype = sizeof(ADataType) * M * K / pack_size_v<ADataType> +
|
||||
sizeof(BDataType) * K * N / pack_size_v<BDataType> +
|
||||
std::size_t num_btype = sizeof(ADataType) * M * K / ck::pack_size_v<ADataType> +
|
||||
sizeof(BDataType) * K * N / ck::pack_size_v<BDataType> +
|
||||
sizeof(CDataType) * M * N +
|
||||
sizeof(XDataType) * (M * K + K * N) / ScaleBlockSize;
|
||||
|
||||
|
||||
Reference in New Issue
Block a user