Skip to content

Commit 391adbc

Browse files
committed
observatory: fix qnn accuracy lens hook
1 parent 35486c1 commit 391adbc

1 file changed

Lines changed: 6 additions & 7 deletions

File tree

backends/qualcomm/debugger/observatory/lenses/qnn_patches.py

Lines changed: 6 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -24,10 +24,10 @@
2424
def install_qnn_patches(cls: type[PipelineGraphCollectorLens]) -> None:
2525
"""Install QNN ptq_calibrate patch on the PipelineGraphCollectorLens."""
2626
try:
27-
import executorch.examples.qualcomm.utils as qnn_utils_module
27+
import executorch.backends.qualcomm.export_utils as qnn_utils_module
2828

29-
original = qnn_utils_module.ptq_calibrate
30-
cls._originals["qnn.ptq_calibrate"] = original
29+
original = qnn_utils_module._ptq_calibrate
30+
cls._originals["qnn._ptq_calibrate"] = original
3131

3232
def patched_ptq_calibrate(captured_model, quantizer, dataset):
3333
cls._set_accuracy_fallback_dataset(
@@ -58,15 +58,14 @@ def patched_ptq_calibrate(captured_model, quantizer, dataset):
5858
"[PipelineGraphCollector] collect skipped (Exported Float): %s",
5959
exc,
6060
)
61-
6261
return original(captured_model, quantizer, dataset)
6362

64-
qnn_utils_module.ptq_calibrate = patched_ptq_calibrate
65-
logging.info("[PipelineGraphCollector] Installed QNN patch: ptq_calibrate")
63+
qnn_utils_module._ptq_calibrate = patched_ptq_calibrate
64+
logging.info("[PipelineGraphCollector] Installed QNN patch: _ptq_calibrate")
6665

6766
def _uninstall():
6867
try:
69-
qnn_utils_module.ptq_calibrate = original
68+
qnn_utils_module._ptq_calibrate = original
7069
except Exception:
7170
pass
7271

0 commit comments

Comments
 (0)