ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-06-07 16:26:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix reference gemm

Browse Source
This commit is contained in:
Rostyslav Geyyer
2025-03-26 14:17:04 +00:00
parent 2fd48e1231
commit 8940acf5ef
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
library/include/ck/library/reference_tensor_operation/cpu/reference_gemm.hpp
View File

@@ -108,10 +108,10 @@ struct ReferenceGemm : public device::BaseOperator
{
if(k % 2 == 1)
v_b = type_convert<ComputeTypeB>(
arg.b_k_n_(k, n).template unpack<>(Number<1>{}));
f4_t{arg.b_k_n_(k, n).template unpack<>(Number<1>{})});
else
v_b = type_convert<ComputeTypeB>(
arg.b_k_n_(k, n).template unpack<>(Number<0>{}));
f4_t{arg.b_k_n_(k, n).template unpack<>(Number<0>{})});
}
else
{