bloqade.rewrite.passes.callgraph.CallGraphPass

bloqade.rewrite.passes.callgraph.CallGraphPass

Bases: passes.Pass

Copy all functions in the call graph and apply a rule to each of them.

Signature

class CallGraphPass(rule: RewriteRule | None = None, rule_factory: Callable[[ir.Method, CallGraph], RewriteRule] | None = None)

Note: This pass modifies the input method in place, but copies all methods invoked within it before applying the rule to them.

Usage

rule = Walk(SomeRewriteRule()) pass_ = CallGraphPass(rule=rule, dialects=…) pass_(some_method)

or, when the rule needs per-method context:

pass_ = CallGraphPass( rule_factory=lambda callee, call_graph: Walk(MyRule(callee, call_graph)), dialects=…, )

Parameters

Name	Type	Default	Description
`rule`	`RewriteRule \| None`	`None`
`rule_factory`	`Callable[[ir.Method, CallGraph], RewriteRule] \| None`	`None`

Attributes

Name	Type	Default	Description
`rule`	`RewriteRule \| None`	`None`	The rule to apply to each function in the call graph.
`rule_factory`	`Callable[[ir.Method, CallGraph], RewriteRule] \| None`	`None`	Build a per-method rule from the callee and call graph. Mutually exclusive with ``rule``.
`fold_pass`	`passes.Fold`	`field(init=False)`

bloqade.rewrite.passes.callgraph.CallGraphPass.unsafe_run

def unsafe_run(mt: ir.Method) -> RewriteResult

Parameters

Name	Type	Description
`mt`	`ir.Method`

Returns

RewriteResult