Skip to content

Commit 10361c4

Browse files
mcr229facebook-github-bot
authored andcommitted
unified partitioner for qs8 mv2 (#306)
Summary: Pull Request resolved: #306 Using Unified Partitioner for MV2 Reviewed By: digantdesai Differential Revision: D49116827 fbshipit-source-id: bbad3b941fb94d05031368fa20f81fcf728b84f4
1 parent 8abb05f commit 10361c4

File tree

1 file changed

+2
-6
lines changed

1 file changed

+2
-6
lines changed

backends/xnnpack/test/models/mobilenet_v2.py

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -8,11 +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
15-
from executorch.exir import CaptureConfig
11+
from executorch.backends.xnnpack.test.tester import Tester
1612
from torchvision.models.mobilenetv2 import MobileNet_V2_Weights
1713

1814

@@ -59,7 +55,7 @@ def test_qs8_mv2(self):
5955
.export()
6056
.to_edge()
6157
.check(list(ops_after_quantization))
62-
.partition(Partition(partitioner=XnnpackQuantizedPartitioner))
58+
.partition()
6359
.check(["torch.ops.executorch_call_delegate"])
6460
.check_not(list(ops_after_quantization))
6561
.to_executorch()

0 commit comments

Comments
 (0)