Skip to content

Commit a6d5711

Browse files
committed
Adjust bf16 test threshold
1 parent 461df5a commit a6d5711

File tree

1 file changed

+5
-0
lines changed
  • src/plugins/intel_cpu/tests/functional/custom/subgraph_tests/src/x64

1 file changed

+5
-0
lines changed

src/plugins/intel_cpu/tests/functional/custom/subgraph_tests/src/x64/moe.cpp

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -87,6 +87,11 @@ class MoESubgraphTest : public testing::WithParamInterface<MoeTestParams>,
8787
init_input_shapes({shape_params.data_shape});
8888
inType = outType = ov::element::f32;
8989

90+
auto itr = configuration.find(ov::hint::inference_precision.name());
91+
if (itr != configuration.end() && itr->second == ov::element::bf16) {
92+
rel_threshold = 0.05f;
93+
}
94+
9095
if (moe_type == MoEType::MoE2GeMM) {
9196
function = initMoE2GeMMSubgraph(shape_params, ov::element::f32, ov::element::f32);
9297
} else if (moe_type == MoEType::MoE3GeMM) {

0 commit comments

Comments
 (0)