bishop: Make call graph generation more robust

stelleg · May 4, 2018 · 3e85004 · 3e85004
1 parent d5eb4f3
commit 3e85004
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/language/src/bishop/optimize_match.t b/language/src/bishop/optimize_match.t
@@ -113,7 +113,8 @@ local function fetch_call_graph()
       if v:has_primary_variant() then
         local task_ast = v:get_primary_variant():get_ast()
         local function record_callees(node)
-          if node:is(regent_ast.typed.expr.Call) and
+          if (node:is(regent_ast.specialized.expr.Call) or
+              node:is(regent_ast.typed.expr.Call)) and
              regent_std.is_task(node.fn.value) then
             call_graph[k][node.fn.value.name:mkstring()] = true
           end