File tree Expand file tree Collapse file tree 1 file changed +2
-4
lines changed
backends/xnnpack/test/tester Expand file tree Collapse file tree 1 file changed +2
-4
lines changed Original file line number Diff line number Diff line change 12
12
import torch
13
13
import torch ._export as export
14
14
from executorch import exir
15
- from executorch .backends .xnnpack .partition .xnnpack_partitioner import (
16
- XnnpackFloatingPointPartitioner ,
17
- )
15
+ from executorch .backends .xnnpack .partition .xnnpack_partitioner import XnnpackPartitioner
18
16
from executorch .backends .xnnpack .passes import XNNPACKPassManager
19
17
from executorch .backends .xnnpack .utils .configs import (
20
18
get_xnnpack_capture_config ,
@@ -184,7 +182,7 @@ def graph_module(self) -> str:
184
182
@register_stage
185
183
class Partition (Stage ):
186
184
def __init__ (self , partitioner : Optional [Partitioner ] = None ):
187
- self .partitioner = partitioner or XnnpackFloatingPointPartitioner
185
+ self .partitioner = partitioner or XnnpackPartitioner
188
186
self .delegate_module = None
189
187
190
188
def run (self , artifact : ExirExportedProgram , inputs = None ):
You can’t perform that action at this time.
0 commit comments