Skip to content

Commit a9d8a65

Browse files
ankalinintprimak
authored andcommitted
cpu: rnn: enable f16 for non-amx isa
1 parent 093c03d commit a9d8a65

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/cpu/x64/rnn/rnn_brgemm_utils.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -89,7 +89,7 @@ x64::cpu_isa_t brgemm_calc_isa(
8989
} else if (rnn.is_cell_dt_bf16()) {
9090
return x64::avx512_core_bf16;
9191
} else if (rnn.is_cell_dt_f16()) {
92-
return isa_undef;
92+
return x64::avx512_core_fp16;
9393
} else { // f32
9494
return utils::map(true, x64::isa_undef, mayiuse(avx512_core),
9595
avx512_core, mayiuse(avx2), avx2);

0 commit comments

Comments
 (0)