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

update

Browse Source
This commit is contained in:
yadaish
2025-12-01 06:59:48 +00:00
parent b2b6fa1aa9
commit b36c7d76a0
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
include/ck_tile/ops/flatmm/pipeline/mixed_prec_flatmm_pipeline_agmem_bgmem_creg_v1.hpp
View File

@@ -288,7 +288,8 @@ struct F16xMXF4FlatmmPipelineAGmemBGmemCRegV1
uint32_t uscale = uint32_t(scale.data) << float_mantissa;
*/
float scale_f32 = type_convert<float>(scale.data);
// float scale_f32 = type_convert<float>(scale.data);
float scale_f32 = type_convert<float>(scale);
using ComputeV2Type =
std::conditional_t<std::is_same_v<ComputeType, half_t>, fp16x2_t, bf16x2_t>;
@@ -303,6 +304,9 @@ struct F16xMXF4FlatmmPipelineAGmemBGmemCRegV1
return pk_int4_t_to_bfloat16x2_t(pk_int4, fscale);
}
else
{
static_assert(sizeof(pk_int4) == 0, "unsupported compute type");
}