M-Labs/artiq
2
0
Fork 2
mirror of https://github.com/m-labs/artiq.git synced 2024-12-25 03:08:27 +08:00
Code Issues Projects Releases Wiki Activity

transforms.artiq_ir_generator: fix devirtualized call annotation.

Browse Source
This commit is contained in:
whitequark 2015-10-14 16:12:00 +03:00
parent e9adfd639e
commit 5d64df829e
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
artiq/compiler/transforms/artiq_ir_generator.py
View File

@ -105,7 +105,9 @@ class ARTIQIRGenerator(algorithm.Visitor):
def annotate_calls(self, devirtualization):
for var_node in devirtualization.variable_map:
callee_node = devirtualization.variable_map[var_node]
callee = self.function_map[callee_node]
if callee_node is None:
continue
callee = self.function_map[callee_node]
call_target = self.variable_map[var_node]
for use in call_target.uses:
@ -115,7 +117,9 @@ class ARTIQIRGenerator(algorithm.Visitor):
for type_and_method in devirtualization.method_map:
callee_node = devirtualization.method_map[type_and_method]
callee = self.function_map[callee_node]
if callee_node is None:
continue
callee = self.function_map[callee_node]
for call in self.method_map[type_and_method]:
assert isinstance(call, (ir.Call, ir.Invoke))