Shape dim could be only torch.SymInt

Author: daniil-lyakhov

nncf/experimental/torch/fx/nncf_graph_builder.py

@@ -198,9 +198,7 @@ def get_edge_params(
             else:
                 tensor = source_node.meta["val"]
             if isinstance(tensor, torch.Tensor):
-                tensor_shape = tuple(
-                    -1 if isinstance(i, GraphConverter.TORCH_SYMBOLIC_TYPES) else i for i in tensor.shape
-                )
+                tensor_shape = tuple(-1 if isinstance(i, torch.SymInt) else i for i in tensor.shape)
             elif isinstance(tensor, GraphConverter.TORCH_SYMBOLIC_TYPES):
                 tensor_shape = (-1,)
 

tests/torch2/function_hook/quantization/test_quantizer_config.py

@@ -15,6 +15,8 @@
 from nncf.common.utils.backend import BackendType
 from nncf.quantization.algorithms.min_max.torch_backend import PTMinMaxAlgoBackend
 from nncf.torch.graph.graph import PTNNCFGraph
+from tests.cross_fw.test_templates.models import NNCFGraphConstantBranchWithWeightedNode
+from tests.cross_fw.test_templates.models import NNCFGraphModelWithEmbeddingsConstantPath
 from tests.cross_fw.test_templates.models import NNCFGraphToTest
 from tests.cross_fw.test_templates.models import NNCFGraphToTestDepthwiseConv
 from tests.cross_fw.test_templates.models import NNCFGraphToTestSumAggregation
@@ -54,3 +56,30 @@ def transformer_nncf_graph(self) -> NNCFGraphToTest:
             transpose_metatype=om.PTTransposeMetatype,
             nncf_graph_cls=PTNNCFGraph,
         )
+
+    @pytest.fixture
+    def embedding_nncf_graph_shape_of(self) -> NNCFGraphToTest:
+        return None
+
+    @pytest.mark.skip("Torch2 does not have shape of subgraphs")
+    def test_embedding_model_qconfig_shape_of(self, embedding_nncf_graph_shape_of):
+        pass
+
+    @pytest.fixture
+    def embedding_nncf_graph_constant_path(self) -> NNCFGraphToTest:
+        return NNCFGraphModelWithEmbeddingsConstantPath(
+            const_metatype=om.PTConstNoopMetatype,
+            embedding_metatype=om.PTModuleEmbeddingMetatype,
+            conv_metatype=om.PTModuleConv2dMetatype,
+            add_metatype=om.PTAddMetatype,
+            nncf_graph_cls=PTNNCFGraph,
+        )
+
+    @pytest.fixture
+    def constant_branch_nncf_graph(self) -> NNCFGraphToTest:
+        return NNCFGraphConstantBranchWithWeightedNode(
+            const_metatype=om.PTConstNoopMetatype,
+            conv_metatype=om.PTModuleConv2dMetatype,
+            add_metatype=om.PTAddMetatype,
+            nncf_graph_cls=PTNNCFGraph,
+        )