Skip to content

Commit c31054c

Browse files
tarun292facebook-github-bot
authored andcommitted
Fix the key used for extraction of delegate debug handle map in emitter
Summary: In D49173918 while looking for the delegate map in the lowered_module metadata i ended up using the wrong key. It should be `debug_handle_map` and not `delegate_map`. Reviewed By: Jack-Khuu Differential Revision: D49620440 fbshipit-source-id: e06bd2ff557eae39173bccbb5fcb10b527bc91bf
1 parent 6c6e27a commit c31054c

File tree

2 files changed

+13
-1
lines changed

2 files changed

+13
-1
lines changed

exir/emit/_emitter.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -950,7 +950,7 @@ def _add_delegate_map(
950950
"""
951951
delegate_map = {}
952952
if hasattr(lowered_module, "meta"):
953-
delegate_map = lowered_module.meta.get("delegate_map", {})
953+
delegate_map = lowered_module.meta.get("debug_handle_map", {})
954954

955955
self.instr_id_to_delegate_debug_id_map[delegate_instruction_id] = {
956956
"name": lowered_module.backend_id,

exir/emit/test/test_emit.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1324,4 +1324,16 @@ def forward(self, x, y):
13241324
.to_edge()
13251325
.to_executorch()
13261326
)
1327+
# Reading the program triggers the call to emit_program underneath which
1328+
# we need to be done for our test to succeed.
1329+
exec_prog.program
13271330
self.assertIsNotNone(exec_prog.delegate_map)
1331+
self.assertIsNotNone(exec_prog.delegate_map.get("forward"))
1332+
self.assertIsNotNone(exec_prog.delegate_map.get("forward").get(0))
1333+
self.assertEqual(
1334+
exec_prog.delegate_map.get("forward").get(0).get("name"),
1335+
"BackendWithCompilerDemo",
1336+
)
1337+
self.assertTrue(
1338+
len(exec_prog.delegate_map.get("forward").get(0).get("delegate_map")) != 0
1339+
)

0 commit comments

Comments
 (0)