Skip to content

Commit 92bf722

Browse files
authored
Fix UnboundLocalError: local variable 'qnn_compile_spec_buffer' refer… (#15881)
As title, addressed issues exposed from pytorch/pytorch#168098 ``` FAILED exir/backend/test/test_debug_handle_map.py::TestBackendDebugHandle::test_lowered_the_whole_model - UnboundLocalError: local variable 'qnn_compile_spec_buffer' referenced before assignment Falsifying example: test_lowered_the_whole_model( unlift=False, self=<test_debug_handle_map.TestBackendDebugHandle testMethod=test_lowered_the_whole_model>, ) ```
1 parent b3b7a98 commit 92bf722

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

backends/qualcomm/partition/utils.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,11 +16,19 @@
1616
def generate_qnn_executorch_option(
1717
compiler_specs: List[CompileSpec],
1818
) -> bytes:
19+
qnn_compile_spec_buffer = None
20+
1921
for compiler_spec in compiler_specs:
2022
if compiler_spec.key == QCOM_QNN_COMPILE_SPEC:
2123
qnn_compile_spec_buffer = compiler_spec.value
2224
else:
2325
raise ValueError(f"unknown compiler spec key value: {compiler_spec.key}")
26+
27+
if qnn_compile_spec_buffer is None:
28+
raise ValueError(
29+
f"QNN compile spec (key={QCOM_QNN_COMPILE_SPEC}) not found in compiler_specs"
30+
)
31+
2432
return qnn_compile_spec_buffer
2533

2634

0 commit comments

Comments
 (0)