Skip to content

Commit

Permalink
force f16 in tests
Browse files Browse the repository at this point in the history
  • Loading branch information
alvoron committed Feb 14, 2024
1 parent 4d8fb9b commit e25d0a7
Showing 1 changed file with 7 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -8,11 +8,17 @@ namespace ov {
namespace test {

void core_configuration(ov::test::SubgraphBaseTest* test) {
#if defined(OV_CPU_ARM_ENABLE_FP16) || defined(OPENVINO_ARCH_X86) || defined(OPENVINO_ARCH_X86_64)
#if defined(OPENVINO_ARCH_X86) || defined(OPENVINO_ARCH_X86_64)
//force fp32 inference precision if it is not configured specially
if (!test->configuration.count(ov::hint::inference_precision.name())) {
test->configuration.insert({ov::hint::inference_precision.name(), ov::element::f32.to_string()});
}
#endif
#if defined(OV_CPU_ARM_ENABLE_FP16)
//force fp16 inference precision if it is not configured specially
if (!test->configuration.count(ov::hint::inference_precision.name())) {
test->configuration.insert({ov::hint::inference_precision.name(), ov::element::f16.to_string()});
}
#endif
// todo: issue: 123320
test->convert_precisions.insert({ov::element::bf16, ov::element::f32});
Expand Down

0 comments on commit e25d0a7

Please sign in to comment.