Skip to content

Commit a80bcb7

Browse files
mcr229facebook-github-bot
authored andcommitted
unified partitione for qs8 mv2
Differential Revision: D49116827 fbshipit-source-id: 49d795b5da3af23ca86e7055bfbb0e9fc9d07af2
1 parent 040856f commit a80bcb7

File tree

1 file changed

+2
-5
lines changed

1 file changed

+2
-5
lines changed

backends/xnnpack/test/models/mobilenet_v2.py

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -8,10 +8,7 @@
88

99
import torch
1010
import torchvision.models as models
11-
from executorch.backends.xnnpack.partition.xnnpack_partitioner import (
12-
XnnpackQuantizedPartitioner,
13-
)
14-
from executorch.backends.xnnpack.test.tester import Export, Partition, Tester
11+
from executorch.backends.xnnpack.test.tester import Export, Tester
1512
from executorch.exir import CaptureConfig
1613
from torchvision.models.mobilenetv2 import MobileNet_V2_Weights
1714

@@ -59,7 +56,7 @@ def test_qs8_mv2(self):
5956
.export()
6057
.to_edge()
6158
.check(list(ops_after_quantization))
62-
.partition(Partition(partitioner=XnnpackQuantizedPartitioner))
59+
.partition()
6360
.check(["torch.ops.executorch_call_delegate"])
6461
.check_not(list(ops_after_quantization))
6562
.to_executorch()

0 commit comments

Comments
 (0)