Skip to content

Commit 4955252

Browse files
mcr229facebook-github-bot
authored andcommitted
Make tester partitioner default partitioner (#300)
Summary: Pull Request resolved: #300 With our new unified partitioner, This should be the default partitioner that we use for testing our end-to-end xnnpack flow. Later when there is more configurability with partitioner, we can modify this to add these configs in testing Reviewed By: manuelcandales Differential Revision: D49214926 fbshipit-source-id: de9fd1b6f67b6fd8faf171679ffbb79e9bcb36d7
1 parent a0e9dba commit 4955252

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)