Skip to content

Commit 3d238ae

Browse files
hsharma35facebook-github-bot
authored andcommitted
Add getitem support in graph builder. (#10910)
Summary: Add call_getitem support in graph builder. Differential Revision: D74817590
1 parent 0a6f622 commit 3d238ae

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

backends/cadence/aot/graph_builder.py

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

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

0 commit comments

Comments
 (0)