File tree Expand file tree Collapse file tree 1 file changed +6
-2
lines changed
backends/xnnpack/partition/config Expand file tree Collapse file tree 1 file changed +6
-2
lines changed Original file line number Diff line number Diff line change 1010from typing import List , Optional
1111
1212import torch
13+ from torch .export import ExportedProgram
14+ from executorch .backends .xnnpack .utils .quant_utils import is_dynamic_qdq
1315from executorch .exir .backend .canonical_partitioners .config_partitioner import (
1416 format_target_name ,
1517 PartitionerConfig ,
1618)
1719from executorch .exir .backend .utils import WhyNoPartition
18- from torch .export import ExportedProgram
1920
2021logger = logging .getLogger (__name__ )
2122why = WhyNoPartition (logger = logger )
@@ -220,9 +221,12 @@ def _check_node_has_valid_dtype(self, node):
220221 valid_dtypes = {
221222 torch .float32 ,
222223 torch .float16 ,
223- torch .int8 ,
224224 torch .qint8 ,
225225 }
226+ # Only allow int8 for quantization operations
227+ if is_dynamic_qdq (node ):
228+ valid_dtypes .add (torch .int8 )
229+
226230 if (
227231 node .op != "placeholder"
228232 and node .op != "call_function"
You can’t perform that action at this time.
0 commit comments