Skip to content

Commit 8bc29a6

Browse files
committed
remove unnecesary FX metatype Registry
1 parent c9c4319 commit 8bc29a6

File tree

2 files changed

+4
-112
lines changed

2 files changed

+4
-112
lines changed

nncf/experimental/torch/fx/nncf_graph_builder.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@
2424
from nncf.experimental.torch.fx.node_utils import get_tensor_constant_from_node
2525
from nncf.torch.dynamic_graph.layer_attributes_handlers import apply_args_defaults
2626
from nncf.torch.graph.graph import PTNNCFGraph
27-
from nncf.torch.graph.operator_metatypes import FX_OPERATOR_METATYPES
27+
from nncf.torch.graph.operator_metatypes import PT_OPERATOR_METATYPES
2828

2929

3030
class GraphConverter:
@@ -94,7 +94,7 @@ def get_node_type_and_metatype(node: torch.fx.Node, model: torch.fx.GraphModule)
9494
else:
9595
# TODO(dlyakhov): get correct nodes types from this nodes as well
9696
node_type = str(node.target)
97-
node_metatype = FX_OPERATOR_METATYPES.get_operator_metatype_by_func(node_type)
97+
node_metatype = PT_OPERATOR_METATYPES.get_operator_metatype_by_func(node_type)
9898
else:
9999
node_type = node.op
100100
node_metatype = UnknownMetatype

0 commit comments

Comments
 (0)