File tree Expand file tree Collapse file tree 2 files changed +4
-2
lines changed
backends/xnnpack/test/recipes Expand file tree Collapse file tree 2 files changed +4
-2
lines changed Original file line number Diff line number Diff line change 18
18
from executorch .examples .models .model_factory import EagerModelFactory
19
19
from executorch .examples .xnnpack import MODEL_NAME_TO_OPTIONS , QuantType
20
20
from executorch .exir .schema import DelegateCall , Program
21
- from executorch .export import export , ExportRecipe
21
+ from executorch .export import export , ExportRecipe , recipe_registry
22
22
from torch import nn
23
23
from torch .testing ._internal .common_quantization import TestHelperModules
24
24
@@ -27,6 +27,7 @@ class TestXnnpackRecipes(unittest.TestCase):
27
27
def setUp (self ) -> None :
28
28
torch ._dynamo .reset ()
29
29
super ().setUp ()
30
+ recipe_registry .register_backend_recipe_provider (XNNPACKRecipeProvider ())
30
31
31
32
def tearDown (self ) -> None :
32
33
super ().tearDown ()
Original file line number Diff line number Diff line change @@ -33,7 +33,8 @@ addopts =
33
33
exir/program/test
34
34
exir/tests/
35
35
# executorch/export
36
- executorch/export/tests
36
+ export/tests
37
+ --ignore =export/tests/test_export_stages.py
37
38
# kernels/
38
39
kernels/prim_ops/test
39
40
kernels/quantized
You can’t perform that action at this time.
0 commit comments