From 0b6b9dbd1b463b83b45f08ae6cf7780f89dd4273 Mon Sep 17 00:00:00 2001 From: mtgu0705 Date: Mon, 22 Sep 2025 02:04:27 -0500 Subject: [PATCH] fix bandwidth calculation --- example/ck_tile/18_flatmm/mxgemm/mx_flatmm.cpp | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/example/ck_tile/18_flatmm/mxgemm/mx_flatmm.cpp b/example/ck_tile/18_flatmm/mxgemm/mx_flatmm.cpp index c358bf043d..6577d8ce6c 100644 --- a/example/ck_tile/18_flatmm/mxgemm/mx_flatmm.cpp +++ b/example/ck_tile/18_flatmm/mxgemm/mx_flatmm.cpp @@ -291,7 +291,9 @@ float invoke_mx_flatmm(ck_tile::DeviceMem& a_dev_buf, std::size_t flop = std::size_t(2) * M * N * K + std::size_t(2) * M * N * K / 32; std::size_t num_byte = sizeof(ADataType) * M * K / APackedSize + - sizeof(BDataType) * N * K / BPackedSize + sizeof(CDataType) * M * N; + sizeof(BDataType) * N * K / BPackedSize + sizeof(CDataType) * M * N + + sizeof(ck_tile::e8m0_t) * M * K / 32 + + sizeof(ck_tile::e8m0_t) * N * K / 32; float tflops = static_cast(flop) / 1.E9 / ave_time; float gb_per_sec = num_byte / 1.E6 / ave_time;