Merge branch 'adrianl/qnn-ep-quant-config' into adrianl/qnn-ep-quant-…

…config-locked
microsoft · Nov 20, 2023 · 001faef · 001faef
2 parents 2c75675 + c452352
commit 001faef
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/onnxruntime/python/tools/quantization/execution_providers/qnn/quant_config.py b/onnxruntime/python/tools/quantization/execution_providers/qnn/quant_config.py
@@ -14,7 +14,6 @@
 Q16_TYPES = {QuantType.QInt16, QuantType.QUInt16}
 Q8_TYPES = {QuantType.QInt8, QuantType.QUInt8}
 OP_TYPES_TO_EXCLUDE = {"Cast"}
-OP_TYPES_INITIALIZER_AS_WEIGHTS = {"MatMul", "Add", "Sub", "Mul", "Div"}
 
 
 def get_qnn_qdq_config(
@@ -35,7 +34,7 @@ def get_qnn_qdq_config(
     for node in model.graph.node:
         op_types.add(node.op_type)
 
-        if node.op_type in OP_TYPES_INITIALIZER_AS_WEIGHTS and activation_type in Q16_TYPES and weight_type in Q8_TYPES:
+        if node.op_type == "MatMul" and activation_type in Q16_TYPES and weight_type in Q8_TYPES:
             weight_symmetric = weight_type == QuantType.QInt8
 
             # Override initializers to use the weight_type