Skip to content

Apply pdl

apply_pdl

ApplyPDLPass dataclass

Bases: ModulePass

Source code in xdsl/transforms/apply_pdl.py 
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
@dataclass(frozen=True)
class ApplyPDLPass(ModulePass):
    name = "apply-pdl"

    pdl_file: str | None = None

    def apply(self, ctx: Context, op: builtin.ModuleOp) -> None:
        if self.pdl_file is not None:
            assert os.path.exists(self.pdl_file)
            with open(self.pdl_file) as f:
                pdl_module_str = f.read()
                parser = Parser(ctx, pdl_module_str)
                pdl_module = parser.parse_module()
        else:
            pdl_module = op
        rewrite_patterns: list[RewritePattern] = [
            PDLRewritePattern(op, ctx, None)
            for op in pdl_module.walk()
            if isinstance(op, pdl.RewriteOp)
        ]
        pattern_applier = GreedyRewritePatternApplier(rewrite_patterns)
        PatternRewriteWalker(pattern_applier).rewrite_module(op)

name = 'apply-pdl' class-attribute instance-attribute

pdl_file: str | None = None class-attribute instance-attribute

__init__(pdl_file: str | None = None) -> None

apply(ctx: Context, op: builtin.ModuleOp) -> None

Source code in xdsl/transforms/apply_pdl.py 
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
def apply(self, ctx: Context, op: builtin.ModuleOp) -> None:
    if self.pdl_file is not None:
        assert os.path.exists(self.pdl_file)
        with open(self.pdl_file) as f:
            pdl_module_str = f.read()
            parser = Parser(ctx, pdl_module_str)
            pdl_module = parser.parse_module()
    else:
        pdl_module = op
    rewrite_patterns: list[RewritePattern] = [
        PDLRewritePattern(op, ctx, None)
        for op in pdl_module.walk()
        if isinstance(op, pdl.RewriteOp)
    ]
    pattern_applier = GreedyRewritePatternApplier(rewrite_patterns)
    PatternRewriteWalker(pattern_applier).rewrite_module(op)