weight dim > 3 assert is removed

Author: daniil-lyakhov

src/nncf/quantization/algorithms/weight_compression/algorithm.py

@@ -1262,11 +1262,6 @@ def get_statistic_points(
                 input_channel_axis = input_channel_axis % n_dims
                 reduction_axes = tuple(i for i in range(n_dims) if i != input_channel_axis)
 
-                if any(weight_dim > 3 for weight_dim in all_weight_dims):
-                    max_val = max(weight_dim for weight_dim in all_weight_dims)
-                    msg = f"Compression with {max_val} dimentional weight is not supported"
-                    raise nncf.InternalError(msg)
-
                 # For 3D weights, keep the batch dimention
                 if any(weight_dim == 3 for weight_dim in all_weight_dims):
                     assert len(reduction_axes) == 2

src/nncf/torch/node_utils.py

@@ -24,7 +24,7 @@ def get_activation_channel_axis(node: NNCFNode, port_id: int) -> int:
     :param port_id: Port ID for input.
     :return: Channel axis number.
     """
-    if node.metatype not in op.CONVOLUTION_METATYPES + op.MATMUL_METATYPES + op.UNIFICATION_PRODUCING_METATYPES:
+    if node.metatype not in op.CONVOLUTION_METATYPES + op.MATMUL_METATYPES:
         msg = f"Activation channel axis retrieval from node with metatype {node.metatype} is not supported"
         raise nncf.InternalError(msg)
 

tests/torch/test_node_utils.py

@@ -21,7 +21,6 @@
     "metatype,port_id,ref_out",
     (
         (op.PTLinearMetatype, 0, -1),
-        (op.PTModuleLinearMetatype, 0, -1),
         (op.PTConv2dMetatype, 0, 1),
         (op.PTDepthwiseConv2dSubtype, 0, 1),
         (op.PTConvTranspose2dMetatype, 0, 1),