From 0501fe46f754572f038516198f3f2688f017b667 Mon Sep 17 00:00:00 2001 From: Radomir Djogo Date: Wed, 30 Oct 2024 16:07:42 +0000 Subject: [PATCH] #14032: change bfp8 pack src format to fp16 --- tt_metal/jit_build/data_format.cpp | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tt_metal/jit_build/data_format.cpp b/tt_metal/jit_build/data_format.cpp index 367d727073d..19bfd3c5926 100644 --- a/tt_metal/jit_build/data_format.cpp +++ b/tt_metal/jit_build/data_format.cpp @@ -334,7 +334,7 @@ const DataFormat get_single_pack_src_format( TT_FATAL(arch != tt::ARCH::GRAYSKULL, "Dest Fp32 mode is not supported for arch grayskull"); if (is_bfp_format(output_format)) { - pack_src_format = DataFormat::Bfp8_b; + pack_src_format = is_exp_b_format(output_format) ? DataFormat::Float16_b : DataFormat::Float16; } else if(is_exp_b_format(output_format) || (output_format == DataFormat::Float32)) { pack_src_format = output_format; } else if(output_format == DataFormat::Float16){ @@ -374,7 +374,7 @@ const DataFormat get_single_pack_src_format( } pack_src_format = unpack_conditional_dst_format; } else if (is_bfp_format(output_format)) { - pack_src_format = is_exp_b_format(output_format) ? DataFormat::Bfp8_b : DataFormat::Bfp8; + pack_src_format = is_exp_b_format(output_format) ? DataFormat::Float16_b : DataFormat::Float16; } else { pack_src_format = output_format; } @@ -390,7 +390,7 @@ const DataFormat get_single_pack_src_format( DataFormat pack_src_format_tmp = output_format; if (is_bfp_format(output_format)) { - pack_src_format_tmp = is_exp_b_format(output_format) ? DataFormat::Bfp8_b : DataFormat::Bfp8; + pack_src_format_tmp = is_exp_b_format(output_format) ? DataFormat::Float16_b : DataFormat::Float16; } if (pack_src_format_tmp != DataFormat::Float32) {