[tracer] fix FUNCTIONAL_MODULE_MAPPING (#357)

alibaba · Sep 18, 2024 · 7cd31e9 · 7cd31e9
1 parent ce644ec
commit 7cd31e9
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/tinynn/graph/quantization/quantizer.py b/tinynn/graph/quantization/quantizer.py
@@ -264,7 +264,6 @@
 
 if hasattr(nn, 'RMSNorm'):
     UNSUPPORTED_PYTORCH_QUANTIZATION_OP_LIST.update({nn.RMSNorm: None})
-    FUNCTIONAL_MODULE_MAPPING.update({'rms_norm': nn.RMSNorm})
 
 # Processed QAT fuse rules
 processed_qat_rules = {}