Skip to content

Commit 4e3c131

Browse files
hsharma35facebook-github-bot
authored andcommitted
Add getitem support in graph builder. (#10910)
Summary: Add call_getitem support in graph builder. Reviewed By: abeakkas Differential Revision: D74817590
1 parent 879235b commit 4e3c131

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

backends/cadence/aot/graph_builder.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -96,6 +96,11 @@ def call_submodule(
9696
) -> PassResult:
9797
return ExportPass().call(graph_module)
9898

99+
def call_getitem(
100+
self, value: ProxyValue, key: int, meta: Optional[NodeMetadata] = None
101+
) -> ProxyValue:
102+
return super().call_getitem(value, key, meta or NodeMetadata({}))
103+
99104
def _fx(
100105
self,
101106
kind: str,

0 commit comments

Comments
 (0)