Skip to content

Commit 1b27c3d

Browse files
committed
cpu: x64: brgemm: enable caching upconverted values for xf16:f8
1 parent 1f8ba64 commit 1b27c3d

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

src/cpu/x64/brgemm/jit_brgemm_amx_uker.cpp

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2227,7 +2227,8 @@ bool jit_brgemm_amx_uker_base_t::maybe_pre_process_data(brgemm_iteration_t &bi,
22272227
// For fp8 via conversion we use temporal buffer heavily for conversion.
22282228
// Therefore saved data may be overwritten
22292229
// TODO: remove this restriction
2230-
if (brg.is_fp8_via_convert()) return false;
2230+
if (brg.is_fp8_via_convert() && !(brg.is_bf16_fp8 || brg.is_f16_fp8))
2231+
return false;
22312232
// save if there is a reuse
22322233
if (mk == matrix_A) {
22332234
return tloop.ldis.size() > 1;

0 commit comments

Comments
 (0)