Skip to content
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.

Commit a86aeaf

Browse files
daniil-lyakhovAlexanderDokuchaev
andauthoredMar 17, 2025··
Apply suggestions from code review
Co-authored-by: Alexander Dokuchaev <alexander.dokuchaev@intel.com>
1 parent 0a2f240 commit a86aeaf

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed
 

‎nncf/quantization/algorithms/min_max/backend.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -301,7 +301,7 @@ def get_weight_nodes(self, inference_nncf_graph: NNCFGraph) -> List[NNCFNode]:
301301
"""
302302
Returns nodes that have weights.
303303
304-
:param nncf_graph: Instance of inference NNCFGraph,
304+
:param inference_nncf_graph: Instance of inference NNCFGraph,
305305
which does not contain shape of and constant subgraphs.
306306
:return: All nodes with weights.
307307
"""

‎nncf/torch/utils.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -473,13 +473,13 @@ def get_model_dtype(model: torch.nn.Module) -> torch.dtype:
473473

474474

475475
def get_weight_nodes_in_inference_grpah(
476-
inference_nncf_graph: NNCFGraph, mat_mul_metatypes: List[om.PTOperatorMetatype]
476+
inference_nncf_graph: NNCFGraph, mat_mul_metatypes: List[Type[om.PTOperatorMetatype]]
477477
) -> List[NNCFNode]:
478478
"""
479479
Returns nodes that have weights.
480480
481481
:param nncf_graph: Instance of inference NNCFGraph,
482-
wich does not contain shape of and constant subgraphs.
482+
which does not contain shape of and constant subgraphs.
483483
:return: All nodes with weights.
484484
"""
485485
weight_nodes_candidates = [
@@ -508,7 +508,7 @@ def is_matmul_with_constant_in_inference_graph(node: NNCFNode, inference_nncf_gr
508508
if node.metatype == om.PTLinearMetatype:
509509
return True
510510

511-
# Inference graph does not containt constans, so
511+
# Inference graph does not contain constants, so
512512
# any missed input edge means it is a constant branch.
513513
return node.metatype in [om.PTMatMulMetatype, om.PTAddmmMetatype] and len(
514514
inference_nncf_graph.get_input_edges(node)

0 commit comments

Comments
 (0)
Please sign in to comment.