ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-07-17 17:19:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

Support A/B Quantization in Blockscale GEMM

Browse Source
This commit is contained in:
KenSCLin
2025-12-04 01:24:06 +00:00
parent f0da8a4be5
commit a3729978ce
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
include/ck_tile/host/reference/reference_gemm.hpp
View File

@@ -177,6 +177,7 @@ CK_TILE_HOST void reference_gemm_abquant(const HostTensor<ADataType>& a_m_k,
else if constexpr(std::is_same_v<BDataType, fp8_t>)
{
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<ADataType>& 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<float>(v_b),static_cast<int>(k)
// ,static_cast<int>(n));
=======
>>>>>>> 198c21436 (Support A/B Quantization in Blockscale GEMM)
}
else
{