Skip to content

Commit

Permalink
LayerNorm metatype was added to ignored metatypes for MinMax
Browse files Browse the repository at this point in the history
  • Loading branch information
daniil-lyakhov committed Oct 4, 2023
1 parent 4260614 commit 9698e6c
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions nncf/quantization/algorithms/min_max/torch_backend.py
Original file line number Diff line number Diff line change
Expand Up @@ -315,6 +315,8 @@ def get_ignored_metatypes(model_type: ModelType, device: TargetDevice) -> List[O
om.PTDivMetatype,
om.PTMaxMetatype,
om.PTSqueezeMetatype,
om.PTLayerNormMetatype,
om.PTModuleLayerNormMetatype,
]
if device != TargetDevice.CPU_SPR:
types.append(om.PTMulMetatype)
Expand Down

0 comments on commit 9698e6c

Please sign in to comment.