Skip to content

Test transform dialect erase schedule

test_transform_dialect_erase_schedule

EraseTransformNamedSequenceOps

Bases: RewritePattern

Source code in xdsl/transforms/test_transform_dialect_erase_schedule.py 
15
16
17
18
19
20
class EraseTransformNamedSequenceOps(RewritePattern):
    @op_type_rewrite_pattern
    def match_and_rewrite(
        self, op: transform.NamedSequenceOp, rewriter: PatternRewriter
    ) -> None:
        rewriter.erase_op(op)

match_and_rewrite(op: transform.NamedSequenceOp, rewriter: PatternRewriter) -> None

Source code in xdsl/transforms/test_transform_dialect_erase_schedule.py 
16
17
18
19
20
@op_type_rewrite_pattern
def match_and_rewrite(
    self, op: transform.NamedSequenceOp, rewriter: PatternRewriter
) -> None:
    rewriter.erase_op(op)

TestTransformDialectEraseSchedulePass dataclass

Bases: ModulePass

Erases transform named sequence operations.

Source code in xdsl/transforms/test_transform_dialect_erase_schedule.py 
23
24
25
26
27
28
29
30
31
32
33
34
35
@dataclass(frozen=True)
class TestTransformDialectEraseSchedulePass(ModulePass):
    """
    Erases transform named sequence operations.
    """

    name = "test-transform-dialect-erase-schedule"

    def apply(self, ctx: Context, op: ModuleOp) -> None:
        PatternRewriteWalker(
            EraseTransformNamedSequenceOps(),
            apply_recursively=False,
        ).rewrite_module(op)

name = 'test-transform-dialect-erase-schedule' class-attribute instance-attribute

__init__() -> None

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

Source code in xdsl/transforms/test_transform_dialect_erase_schedule.py 
31
32
33
34
35
def apply(self, ctx: Context, op: ModuleOp) -> None:
    PatternRewriteWalker(
        EraseTransformNamedSequenceOps(),
        apply_recursively=False,
    ).rewrite_module(op)