Skip to content

Commit c878c78

Browse files
committed
Qualcomm AI Engine Direct - fixed missing import in custom op example
1 parent 96aeb57 commit c878c78

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/qualcomm/custom_op/custom_ops_1.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -84,7 +84,7 @@ def annotate_custom(gm: torch.fx.GraphModule) -> None:
8484
continue
8585

8686
# skip annotation if it is already annotated
87-
if _is_annotated([node]):
87+
if Q_ANNOTATION_KEY in node.meta and node.meta[Q_ANNOTATION_KEY]._annotated:
8888
continue
8989

9090
input_qspec_map = {}

0 commit comments

Comments
 (0)