mark EbmerddingBag with q8_w_sym_any_nr

openvinotoolkit · Jan 31, 2025 · 403fb59 · 403fb59
1 parent cfedaf7
commit 403fb59
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 19 deletions.
diff --git a/nncf/common/hardware/configs/npu.json b/nncf/common/hardware/configs/npu.json
@@ -408,7 +408,7 @@
             "type": "EmbeddingBag",
             "quantization": {
                 "weights": [
-                    "q8_w_sym", "q8_w_asym"
+                    "q8_w_sym_any_nr", "q8_w_asym"
                 ]
             }
         },

diff --git a/tests/cross_fw/shared/json.py b/tests/cross_fw/shared/json.py
@@ -38,8 +38,6 @@ def default(self, o):
             return int(o)
         if isinstance(o, np.floating):
             return float(o)
-        if isinstance(o, Tensor):
-            o = o.data
         if isinstance(o, np.ndarray):
             return o.tolist()
         if isinstance(o, torch.Tensor):

diff --git a/tests/cross_fw/test_templates/fq_params/fq_params.json b/tests/cross_fw/test_templates/fq_params/fq_params.json
@@ -1,10 +1,24 @@
 {
+    "weights_symmetric_sign_True_per_ch_False_narrow_range_True_hf_range_True": {
+        "levels": 255,
+        "input_low": -1.994419813156128,
+        "input_high": 1.994419813156128,
+        "output_low": -1.994419813156128,
+        "output_high": 1.994419813156128
+    },
+    "weights_symmetric_sign_True_per_ch_False_narrow_range_True_hf_range_False": {
+        "levels": 255,
+        "input_low": -0.997209906578064,
+        "input_high": 0.997209906578064,
+        "output_low": -0.997209906578064,
+        "output_high": 0.997209906578064
+    },
     "weights_symmetric_sign_True_per_ch_False_narrow_range_False_hf_range_True": {
         "levels": 256,
         "input_low": -2.0340540409088135,
-        "input_high": 2.002271890640259,
+        "input_high": 2.0022718596646167,
         "output_low": -2.0340540409088135,
-        "output_high": 2.002271890640259
+        "output_high": 2.0022718596646167
     },
     "weights_symmetric_sign_True_per_ch_False_narrow_range_False_hf_range_False": {
         "levels": 256,
@@ -242,20 +256,6 @@
             ]
         ]
     },
-    "weights_symmetric_sign_True_per_ch_False_narrow_range_True_hf_range_True": {
-        "levels": 255,
-        "input_low": -1.994419813156128,
-        "input_high": 1.994419813156128,
-        "output_low": -1.994419813156128,
-        "output_high": 1.994419813156128
-    },
-    "weights_symmetric_sign_True_per_ch_False_narrow_range_True_hf_range_False": {
-        "levels": 255,
-        "input_low": -0.997209906578064,
-        "input_high": 0.997209906578064,
-        "output_low": -0.997209906578064,
-        "output_high": 0.997209906578064
-    },
     "weights_asymmetric_sign_None_per_ch_False_narrow_range_True_hf_range_False": {
         "levels": 255,
         "input_low": 0.0,