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

Merge commit '47e2ed838e3547bba1b48d3f559f20f46fd07b87' into develop

Browse Source
This commit is contained in:
assistant-librarian[bot]
2025-11-20 02:43:03 +00:00
parent ca48bf3b98
commit 809c1ead72
183 changed files with 987 additions and 863 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -432,6 +432,12 @@ CK_TILE_HOST void reference_mx_gemm(const HostTensor<ADataType>& a_m_k,
a_m_k_scaled(m, k) = a_f4_lo * a_scale;
a_m_k_scaled(m, k + 1) = a_f4_hi * a_scale;
}
else
{
a_m_k_scaled(m, k) =
ck_tile::type_convert<AccDataType>((a_m_k(m, k))) *
ck_tile::type_convert<AccDataType>(scale_a(m, k / ScaleBlockSize));
}
}
}