Skip to content

Rewriter

oqd_compiler_infrastructure.rewriter

RewriterBase

Bases: PassBase

This class represents a wrapper for passes to compose and modify their logic without affecting the internals of a pass.

Acknowledgement

This code was inspired by SynbolicUtils.jl, Liang.jl.

Source code in oqd-compiler-infrastructure/src/oqd_compiler_infrastructure/rewriter.py 
28
29
30
31
32
33
34
35
36
37
class RewriterBase(PassBase):
    """
    This class represents a wrapper for passes to compose and modify their logic without
    affecting the internals of a pass.

    Acknowledgement:
        This code was inspired by [SynbolicUtils.jl](https://github.com/JuliaSymbolics/SymbolicUtils.jl/blob/master/src/rewriters.jl), [Liang.jl](https://github.com/Roger-luo/Liang.jl/tree/main/src/rewrite).
    """

    pass

Chain

Bases: RewriterBase

This class represents a composite pass where the passes are applied sequentially.

Acknowledgement

This code was inspired by SymbolicUtils.jl, Liang.jl.

Source code in oqd-compiler-infrastructure/src/oqd_compiler_infrastructure/rewriter.py 
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
class Chain(RewriterBase):
    """
    This class represents a composite pass where the passes are applied sequentially.

    Acknowledgement:
        This code was inspired by [SymbolicUtils.jl](https://github.com/JuliaSymbolics/SymbolicUtils.jl/blob/master/src/rewriters.jl#L64C8-L64C13), [Liang.jl](https://github.com/Roger-luo/Liang.jl/blob/main/src/rewrite/chain.jl).
    """

    def __init__(self, *rules):
        super().__init__()

        self.rules = list(rules)
        pass

    @property
    def children(self):
        return self.rules

    def map(self, model):
        new_model = model
        for rule in self.rules:
            new_model = rule(new_model)
        return new_model

FixedPoint

Bases: RewriterBase

This class represents a wrapped pass that is applied until the object/IR converges to a fixed point or reaches a maximum iteration count.

Acknowledgement

This code was inspired by SymbolicUtils.jl, Liang.jl.

Source code in oqd-compiler-infrastructure/src/oqd_compiler_infrastructure/rewriter.py 
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
class FixedPoint(RewriterBase):
    """
    This class represents a wrapped pass that is applied until the object/IR converges to a fixed point
    or reaches a maximum iteration count.

    Acknowledgement:
        This code was inspired by [SymbolicUtils.jl](https://github.com/JuliaSymbolics/SymbolicUtils.jl/blob/master/src/rewriters.jl#L117C8-L117C16), [Liang.jl](https://github.com/Roger-luo/Liang.jl/blob/main/src/rewrite/fixpoint.jl).
    """

    def __init__(self, rule, *, max_iter=1000):
        super().__init__()

        self.rule = rule
        self.max_iter = max_iter
        pass

    @property
    def children(self):
        return [self.rule]

    def map(self, model):
        i = 0
        new_model = model
        while True:
            _model = self.rule(new_model)

            if _model == new_model or i >= self.max_iter:
                return new_model

            new_model = _model
            i += 1