[GNA] Make PWL unit tests take into account saturation errata

openvinotoolkit · Apr 30, 2021 · d742be2 · d742be2
1 parent f08e6ea
commit d742be2
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 4 deletions.
diff --git a/inference-engine/tests_deprecated/unit/engines/gna/gna_pwl_test.cpp b/inference-engine/tests_deprecated/unit/engines/gna/gna_pwl_test.cpp
@@ -147,7 +147,7 @@ TEST_F(PWLAproximationTest, forReLUonRecursiveAlgoWithSegmentThresholdIsSuccess)
                                 .propagate_forward()
                                 .called_with()
                                 .pwl_quantization_activation(DnnActivationType::kActRelu)
-                                .pwl_quantization_segments_threshold(2);
+                                .pwl_quantization_segments_threshold(4);
 }
 
 TEST_F(PWLAproximationTest, forLeakyReLUonRecursiveAlgoWithSegmentThresholdIsSuccess) {
@@ -157,7 +157,7 @@ TEST_F(PWLAproximationTest, forLeakyReLUonRecursiveAlgoWithSegmentThresholdIsSuc
                                 .propagate_forward()
                                 .called_with()
                                 .pwl_quantization_activation(DnnActivationType::kActLeakyRelu)
-                                .pwl_quantization_segments_threshold(2);
+                                .pwl_quantization_segments_threshold(4);
 }
 
 TEST_F(PWLAproximationTest, DISABLED_forIdentityOnRecursiveAlgoWithSegmentThresholdIsSuccess) {

diff --git a/inference-engine/tests_deprecated/unit/engines/gna/matchers/pwl_matcher.hpp b/inference-engine/tests_deprecated/unit/engines/gna/matchers/pwl_matcher.hpp
@@ -102,8 +102,14 @@ class PWLMatcher : public ::testing::MatcherInterface<const gna_nnet_type_t*> {
         }
 
         switch (slopeChangedTimes) {
-            case 2 : return kActRelu; // also relu has y=0 segment while identity doenst have
-            case 3 : return kActIdentity;
+            case 3 :
+                if (comp.op.pwl.num_segments == 4) {
+                    // ReLU has y=0 segment while identity doesn't have
+                    // 2 segments are added: one at the begining and one at the end, due to saturation errata
+                    return kActRelu;
+                } else {
+                    return kActIdentity;
+                }
             default:
                 // currently cannot determine between sigmoid or tanh etc
                 if (slopeChangedTimes > 3) {