Skip to content

Commit dc97f65

Browse files
committed
check qnn graph
1 parent 774d37e commit dc97f65

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

.ci/scripts/test_wheel_package_qnn.sh

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@ from executorch.backends.qualcomm.utils.utils import (
2525
get_soc_to_chipset_map,
2626
to_edge_transform_and_lower_to_qnn,
2727
)
28+
from executorch.exir.backend.utils import format_delegated_graph
2829
from executorch.examples.models.model_factory import EagerModelFactory
2930
from executorch.exir.capture._config import ExecutorchBackendConfig
3031
from executorch.extension.export_util.utils import save_pte_program
@@ -70,6 +71,9 @@ def main() -> None:
7071
backend_options=backend_options,
7172
)
7273
delegated_program = to_edge_transform_and_lower_to_qnn(m, example_inputs, compile_spec)
74+
output_graph = format_delegated_graph(delegated_program.exported_program().graph_module)
75+
# Ensure QnnBackend is in the output graph
76+
assert "QnnBackend" in output
7377
executorch_program = delegated_program.to_executorch(
7478
config=ExecutorchBackendConfig(extract_delegate_segments=False)
7579
)

0 commit comments

Comments
 (0)