Skip to content

Commit be3eb3e

Browse files
mcr229facebook-github-bot
authored andcommitted
unified partitione for qs8 mv2
Differential Revision: D49116827 fbshipit-source-id: b24a335134aa59afb282e23eaa422de59cae7514
1 parent 4f5a700 commit be3eb3e

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)