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 1818from executorch .examples .models .model_factory import EagerModelFactory
1919from executorch .examples .xnnpack import MODEL_NAME_TO_OPTIONS , QuantType
2020from executorch .exir .schema import DelegateCall , Program
21- from executorch .export import export , ExportRecipe
21+ from executorch .export import export , ExportRecipe , recipe_registry
2222from torch import nn
2323from torch .testing ._internal .common_quantization import TestHelperModules
2424
@@ -27,6 +27,7 @@ class TestXnnpackRecipes(unittest.TestCase):
2727 def setUp (self ) -> None :
2828 torch ._dynamo .reset ()
2929 super ().setUp ()
30+ recipe_registry .register_backend_recipe_provider (XNNPACKRecipeProvider ())
3031
3132 def tearDown (self ) -> None :
3233 super ().tearDown ()
Original file line number Diff line number Diff line change @@ -33,7 +33,8 @@ addopts =
3333 exir/program/test
3434 exir/tests/
3535 # executorch/export
36- executorch/export/tests
36+ export/tests
37+ --ignore =export/tests/test_export_stages.py
3738 # kernels/
3839 kernels/prim_ops/test
3940 kernels/quantized
You can’t perform that action at this time.
0 commit comments