From 9395318666802fc2ac9a3d2be2bd323fc33e5372 Mon Sep 17 00:00:00 2001 From: Andriy Roshchenko <107577548+andriy-ca@users.noreply.github.com> Date: Fri, 11 Jul 2025 13:07:05 -0600 Subject: [PATCH] MX GEMM - FP6 Support in GEMM MX v3 Pipeline (#2481) * Add GEMM MX BF6 example * Fix BF6 type_convert * Add type_convert for bf16x6 * Add compare operator to f4x2_pk_t * Update README for 67_gemm_microscaling * Fix host tensor initialization with integer values for FP8 [ROCm/composable_kernel commit: 3421272f90a64610081af948220ee54fc1c599b7] --- include/ck/utility/data_type.hpp | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/include/ck/utility/data_type.hpp b/include/ck/utility/data_type.hpp index 5fbe30d21b..ff611d2136 100644 --- a/include/ck/utility/data_type.hpp +++ b/include/ck/utility/data_type.hpp @@ -71,6 +71,17 @@ struct f4x2_pk_t { return !(lhs == rhs); } + + // Compare operator + __host__ __device__ friend bool operator==(const f4x2_pk_t& lhs, const f4x2_pk_t& rhs) + { + return lhs.data == rhs.data; + } + + __host__ __device__ friend bool operator!=(const f4x2_pk_t& lhs, const f4x2_pk_t& rhs) + { + return !(lhs == rhs); + } }; template