Skip to content

Commit

Permalink
adjust tests threshold on arm
Browse files Browse the repository at this point in the history
  • Loading branch information
alvoron committed Feb 15, 2025
1 parent 0eaeda4 commit c9809be
Show file tree
Hide file tree
Showing 4 changed files with 12 additions and 3 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -119,9 +119,9 @@
#include "low_precision/fold_convert.hpp"
#include "low_precision/fuse_convert.hpp"
#include "low_precision/group_convolution.hpp"
#include "low_precision/mat_mul.hpp"
#include "low_precision/multiply_to_group_convolution.hpp"
#include "low_precision/network_helper.hpp"
#include "low_precision/mat_mul.hpp"
#include "low_precision/recurrent_cell.hpp"
#include "low_precision/rt_info/bias_attribute.hpp"
#include "transformations/low_precision/mark_dequantization_subgraph.hpp"
Expand Down Expand Up @@ -848,7 +848,7 @@ void Transformations::Lpt(const std::vector<ov::element::Type>& defaultPrecision
[&](const_node_ptr& node) -> bool {
if (NetworkHelper::isConstantPath(node->get_input_node_shared_ptr(1)) &&
one_of(node->input_value(1).get_partial_shape().rank().get_length(), 2, 3)) {
return false;
return false;
}
return true;
},
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -119,8 +119,11 @@ class ConvsAndSums : virtual public SubgraphBaseStaticTest {

auto result = std::make_shared<ov::op::v0::Result>(relu3);
function = std::make_shared<ov::Model>(result, params, "SimpleNet");

#if defined(OPENVINO_ARCH_ARM64) || defined(OPENVINO_ARCH_ARM)
abs_threshold = 6e-3;
#else
abs_threshold = 9e-4;
#endif
}
};

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -64,6 +64,9 @@ void FakeQuantizeLayerTest::SetUp() {
if (fq_direct_arg.size() != 0) {
abs_threshold = (fq_direct_arg[3] - fq_direct_arg[2]) / levels;
}
#if defined(OPENVINO_ARCH_ARM64) || defined(OPENVINO_ARCH_ARM)
abs_threshold = 7e-2;
#endif
auto param = std::make_shared<ov::op::v0::Parameter>(model_type, inputDynamicShapes.front());

std::shared_ptr<ov::Node> fq;
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -56,6 +56,9 @@ void IntegerReduceMeanTest::SetUp() {
} else {
reduce_mean = std::make_shared<ov::op::v1::ReduceMean>(dataNode, axesNode, true);
}
#if defined(OPENVINO_ARCH_ARM64) || defined(OPENVINO_ARCH_ARM)
abs_threshold = 2e-1;
#endif

ov::ParameterVector inputs;
inputs.push_back(dataNode);
Expand Down

0 comments on commit c9809be

Please sign in to comment.