smoke test use xnnpack

pytorch · Jul 25, 2024 · c569f54 · c569f54
1 parent 3c321e3
commit c569f54
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/build/packaging/smoke_test.py b/build/packaging/smoke_test.py
@@ -26,6 +26,7 @@
 import tempfile
 from typing import Tuple
 
+from executorch.backends.xnnpack.partition.xnnpack_partitioner import XnnpackPartitioner
 from executorch.exir import to_edge
 from torch.export import export
 
@@ -60,8 +61,10 @@ def export_linear_model() -> bytes:
     exported_program = export(LinearModel(), example_inputs)
     print("Lowering to edge...")
     edge_program = to_edge(exported_program)
+    print("Lowering to XNNPACK...")
+    edge_delegated = edge.to_backend(XnnpackPartitioner())
     print("Creating ExecuTorch program...")
-    et_program = edge_program.to_executorch()
+    et_program = edge_delegated.to_executorch()
 
     return et_program.buffer