diff --git a/include/ck_tile/core/numeric/pk_int4.hpp b/include/ck_tile/core/numeric/pk_int4.hpp index f0a681cb17..47297b1ef2 100644 --- a/include/ck_tile/core/numeric/pk_int4.hpp +++ b/include/ck_tile/core/numeric/pk_int4.hpp @@ -180,7 +180,7 @@ CK_TILE_HOST_DEVICE bf16x2_t pk_int4_t_to_bfloat16x2_t(const pk_int4_t& x) CK_TILE_HOST_DEVICE bf16x2_t pk_int4_t_to_bfloat16x2_t(const pk_int4_t& x, float scale) { - auto float_vec2 = pk_int4_t_to_fp32x2_t(x); + auto float_vec2 = pk_int4_t_to_fp32x2_t_signed_conversion(x); float_vec2.x = float_vec2.x * scale; float_vec2.y = float_vec2.y * scale; return bf16x2_t{type_convert(float_vec2.x), type_convert(float_vec2.y)};