[CPU][ARM] Fix inference precision for behaviour tests (#19485)
This commit is contained in:
parent
e8f1df495c
commit
9b10ef6f6f
@ -156,11 +156,7 @@ TEST_F(OVClassConfigTestCPU, smoke_PluginSetConfigAffinity) {
|
|||||||
TEST_F(OVClassConfigTestCPU, smoke_PluginSetConfigHintInferencePrecision) {
|
TEST_F(OVClassConfigTestCPU, smoke_PluginSetConfigHintInferencePrecision) {
|
||||||
ov::Core ie;
|
ov::Core ie;
|
||||||
auto value = ov::element::f32;
|
auto value = ov::element::f32;
|
||||||
#if defined(OV_CPU_ARM_ENABLE_FP16)
|
|
||||||
const auto precision = ov::element::f16;
|
|
||||||
#else
|
|
||||||
const auto precision = InferenceEngine::with_cpu_x86_bfloat16() ? ov::element::bf16 : ov::element::f32;
|
const auto precision = InferenceEngine::with_cpu_x86_bfloat16() ? ov::element::bf16 : ov::element::f32;
|
||||||
#endif
|
|
||||||
|
|
||||||
ASSERT_NO_THROW(value = ie.get_property("CPU", ov::hint::inference_precision));
|
ASSERT_NO_THROW(value = ie.get_property("CPU", ov::hint::inference_precision));
|
||||||
ASSERT_EQ(precision, value);
|
ASSERT_EQ(precision, value);
|
||||||
@ -194,11 +190,7 @@ TEST_F(OVClassConfigTestCPU, smoke_PluginSetConfigEnableProfiling) {
|
|||||||
ASSERT_EQ(enableProfiling, value);
|
ASSERT_EQ(enableProfiling, value);
|
||||||
}
|
}
|
||||||
|
|
||||||
#if defined(OV_CPU_ARM_ENABLE_FP16)
|
|
||||||
const auto expected_precision_for_performance_mode = ov::element::f16;
|
|
||||||
#else
|
|
||||||
const auto expected_precision_for_performance_mode = InferenceEngine::with_cpu_x86_bfloat16() ? ov::element::bf16 : ov::element::f32;
|
const auto expected_precision_for_performance_mode = InferenceEngine::with_cpu_x86_bfloat16() ? ov::element::bf16 : ov::element::f32;
|
||||||
#endif
|
|
||||||
|
|
||||||
const auto bf16_if_can_be_emulated = InferenceEngine::with_cpu_x86_avx512_core() ? ov::element::bf16 : ov::element::f32;
|
const auto bf16_if_can_be_emulated = InferenceEngine::with_cpu_x86_avx512_core() ? ov::element::bf16 : ov::element::f32;
|
||||||
using ExpectedModeAndType = std::pair<ov::hint::ExecutionMode, ov::element::Type>;
|
using ExpectedModeAndType = std::pair<ov::hint::ExecutionMode, ov::element::Type>;
|
||||||
|
Loading…
Reference in New Issue
Block a user