Skip to content

Commit bbb231f

Browse files
[CPU] Fixed BF16 Matmul inference precision
1 parent 160af2e commit bbb231f

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/plugins/intel_cpu/src/nodes/executors/fullyconnected_implementations.cpp

+2-2
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@ static const LayoutConfig dnnlFCLayoutConfig{LayoutType::ncsp, LayoutType::ncsp,
4040
// clang-format off
4141
static const TypeMapping dnnlFCTypeMapping {
4242
// {src, wei, bia, dst} pt<src, wei, bias, dst>
43-
{{_bf16, _bf16, _any, _bf16 | _f32}, pt(bypass(), bypass(), use<3>(), use<3>())},
43+
{{_bf16, _bf16 | _f32, _any, _bf16 | _f32}, pt(bypass(), bypass(), use<3>(), use<3>())},
4444
{{_f16, _f16, _any, _f16 | _f32}, pt(bypass(), bypass(), use<3>(), use<3>())},
4545
// integer precision outputs are not supported for float precision inputs
4646
{{_f32 | _bf16 | _f16, _any, _any, _i8 | _u8}, pt(bypass(), bypass(), use<0>(), use<0>())},
@@ -63,7 +63,7 @@ static const MappingNotation dnnlConvolutionMappingNotation {
6363

6464
static const TypeMapping dnnlConvolutionTypeMapping {
6565
// {src, wei, bia, dst} pt<src, wei, bias, dst>
66-
{{_bf16, _bf16, _any, _bf16 | _f32}, pt(bypass(), bypass(), use<3>(), use<3>())},
66+
{{_bf16, _bf16 | _f32, _any, _bf16 | _f32}, pt(bypass(), bypass(), use<3>(), use<3>())},
6767
{{_f16, _f16, _any, _f16 | _f32}, pt(bypass(), bypass(), use<3>(), use<3>())},
6868
// integer precision outputs are not supported for float precision inputs
6969
{{_f32 | _bf16 | _f16, _any, _any, _i8 | _u8}, pt(bypass(), bypass(), use<0>(), use<0>())},

0 commit comments

Comments
 (0)