Linalg generalize named ops

linalg_generalize_named_ops

GeneralizeNamedOpPattern

Bases: RewritePattern

Source code in xdsl/transforms/linalg_generalize_named_ops.py

class GeneralizeNamedOpPattern(RewritePattern):
    @op_type_rewrite_pattern
    def match_and_rewrite(
        self, op: linalg.NamedOperation, rewriter: PatternRewriter
    ) -> None:
        """
        Rewrite a named linalg op to `linalg.generic`.
        """

        generic = linalg.GenericOp(
            op.inputs,
            op.outputs,
            op.body.clone(),
            op.get_indexing_maps(),
            op.get_iterator_types(),
            op.results.types,
        )
        rewriter.replace_op(op, generic)

match_and_rewrite(op: linalg.NamedOperation, rewriter: PatternRewriter) -> None

Rewrite a named linalg op to linalg.generic.

Source code in xdsl/transforms/linalg_generalize_named_ops.py

@op_type_rewrite_pattern
def match_and_rewrite(
    self, op: linalg.NamedOperation, rewriter: PatternRewriter
) -> None:
    """
    Rewrite a named linalg op to `linalg.generic`.
    """

    generic = linalg.GenericOp(
        op.inputs,
        op.outputs,
        op.body.clone(),
        op.get_indexing_maps(),
        op.get_iterator_types(),
        op.results.types,
    )
    rewriter.replace_op(op, generic)

LinalgGeneralizeNamedOpsPass dataclass

Bases: ModulePass

Converts linalg named ops to linalg.generic.

Source code in xdsl/transforms/linalg_generalize_named_ops.py

@dataclass(frozen=True)
class LinalgGeneralizeNamedOpsPass(ModulePass):
    """
    Converts linalg named ops to linalg.generic.
    """

    name = "linalg-generalize-named-ops"

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

name = 'linalg-generalize-named-ops' class-attribute instance-attribute

__init__() -> None

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

Source code in xdsl/transforms/linalg_generalize_named_ops.py

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