From 95c52295c055abb456a6ce409f81501d28465cef Mon Sep 17 00:00:00 2001 From: rahjain-amd Date: Tue, 15 Jul 2025 21:25:56 +0530 Subject: [PATCH] add missing condition for bf16 (#2502) Without this DataType = unknown - ``` sh Run Flatmm kernel with DataType = unknown M =1280 N =16384 K =1024 StrideA =1024 StrideB =1024 StrideC =16384 : 0.228837 ms, 187.687 TFlops, 341.374 GB/s, ``` after this change ```sh Run Flatmm kernel with DataType = bf16 M =1280 N =16384 K =1024 StrideA =1024 StrideB =1024 StrideC =16384 : 0.227029 ms, 189.181 TFlops, 344.092 GB/s, ``` [ROCm/composable_kernel commit: 6b09f0823e64df0bcd035443c9ce4599a838de02] --- example/ck_tile/18_flatmm/run_flatmm_example.inc | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/example/ck_tile/18_flatmm/run_flatmm_example.inc b/example/ck_tile/18_flatmm/run_flatmm_example.inc index 3c4d0908dd..b583612cfb 100644 --- a/example/ck_tile/18_flatmm/run_flatmm_example.inc +++ b/example/ck_tile/18_flatmm/run_flatmm_example.inc @@ -18,6 +18,10 @@ constexpr const char* DataTypeToString() { return "bf8"; } + else if constexpr(std::is_same_v) + { + return "bf16"; + } else { return "unknown";