X86 infer broadcast

x86_infer_broadcast

DS_VpbroadcastqOpScalarLoad

Bases: RewritePattern

Source code in xdsl/transforms/x86_infer_broadcast.py

class DS_VpbroadcastqOpScalarLoad(RewritePattern):
    @op_type_rewrite_pattern
    def match_and_rewrite(
        self, op: x86.DS_VpbroadcastqOp, rewriter: PatternRewriter
    ) -> None:
        if isinstance(mov_op := op.source.owner, x86.ops.DM_MovOp):
            rewriter.replace_op(
                op,
                x86.DM_VbroadcastsdOp(
                    mov_op.memory, mov_op.memory_offset, destination=op.destination.type
                ),
            )

match_and_rewrite(op: x86.DS_VpbroadcastqOp, rewriter: PatternRewriter) -> None

Source code in xdsl/transforms/x86_infer_broadcast.py

@op_type_rewrite_pattern
def match_and_rewrite(
    self, op: x86.DS_VpbroadcastqOp, rewriter: PatternRewriter
) -> None:
    if isinstance(mov_op := op.source.owner, x86.ops.DM_MovOp):
        rewriter.replace_op(
            op,
            x86.DM_VbroadcastsdOp(
                mov_op.memory, mov_op.memory_offset, destination=op.destination.type
            ),
        )

X86InferBroadcast dataclass

Bases: ModulePass

Rewrites a scalar load + broadcast to a broadcast load operation.

Source code in xdsl/transforms/x86_infer_broadcast.py

@dataclass(frozen=True)
class X86InferBroadcast(ModulePass):
    """
    Rewrites a scalar load + broadcast to a broadcast load operation.
    """

    name = "x86-infer-broadcast"

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

name = 'x86-infer-broadcast' class-attribute instance-attribute

__init__() -> None

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

Source code in xdsl/transforms/x86_infer_broadcast.py

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