From a3729978ced43f07a93f4683d87fa4c1173f68d5 Mon Sep 17 00:00:00 2001 From: KenSCLin Date: Thu, 4 Dec 2025 01:24:06 +0000 Subject: [PATCH] Support A/B Quantization in Blockscale GEMM --- include/ck_tile/host/reference/reference_gemm.hpp | 3 +++ 1 file changed, 3 insertions(+) diff --git a/include/ck_tile/host/reference/reference_gemm.hpp b/include/ck_tile/host/reference/reference_gemm.hpp index d67a747977..622eb74f67 100644 --- a/include/ck_tile/host/reference/reference_gemm.hpp +++ b/include/ck_tile/host/reference/reference_gemm.hpp @@ -177,6 +177,7 @@ CK_TILE_HOST void reference_gemm_abquant(const HostTensor& a_m_k, else if constexpr(std::is_same_v) { v_b = fp8_to_float_raw(b_element_op(b_k_n(k, n))); +<<<<<<< HEAD } else { @@ -306,6 +307,8 @@ CK_TILE_HOST void reference_gemm_abquant(const HostTensor& a_m_k, v_b = fp8_to_float_raw(b_element_op(b_k_n(k, n))); // printf("B %f k=%d n=%d\n", static_cast(v_b),static_cast(k) // ,static_cast(n)); +======= +>>>>>>> 198c21436 (Support A/B Quantization in Blockscale GEMM) } else {