Skip to content

Commit 1f2052a

Browse files
mcr229facebook-github-bot
authored andcommitted
Make tester partitioner default partitioner
Differential Revision: D49214926 fbshipit-source-id: 3fb6f64a38e3e7ecc73cb063c906e2d0fbcc6b99
1 parent e67185b commit 1f2052a

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

backends/xnnpack/test/tester/tester.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -12,9 +12,7 @@
1212
import torch
1313
import torch._export as export
1414
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
1816
from executorch.backends.xnnpack.passes import XNNPACKPassManager
1917
from executorch.backends.xnnpack.utils.configs import (
2018
get_xnnpack_capture_config,
@@ -184,7 +182,7 @@ def graph_module(self) -> str:
184182
@register_stage
185183
class Partition(Stage):
186184
def __init__(self, partitioner: Optional[Partitioner] = None):
187-
self.partitioner = partitioner or XnnpackFloatingPointPartitioner
185+
self.partitioner = partitioner or XnnpackPartitioner
188186
self.delegate_module = None
189187

190188
def run(self, artifact: ExirExportedProgram, inputs=None):

0 commit comments

Comments
 (0)