Convert vector to ptr

convert_vector_to_ptr

VectorStoreToPtr dataclass

Bases: RewritePattern

Source code in xdsl/transforms/convert_vector_to_ptr.py

@dataclass
class VectorStoreToPtr(RewritePattern):
    @op_type_rewrite_pattern
    def match_and_rewrite(self, op: vector.StoreOp, rewriter: PatternRewriter):
        assert isa(memref_type := op.base.type, memref.MemRefType)
        target_ptr = get_target_ptr(op.base, memref_type, op.indices, rewriter)
        rewriter.replace_op(op, ptr.StoreOp(addr=target_ptr, value=op.vector))

__init__() -> None

match_and_rewrite(op: vector.StoreOp, rewriter: PatternRewriter)

Source code in xdsl/transforms/convert_vector_to_ptr.py

@op_type_rewrite_pattern
def match_and_rewrite(self, op: vector.StoreOp, rewriter: PatternRewriter):
    assert isa(memref_type := op.base.type, memref.MemRefType)
    target_ptr = get_target_ptr(op.base, memref_type, op.indices, rewriter)
    rewriter.replace_op(op, ptr.StoreOp(addr=target_ptr, value=op.vector))

VectorLoadToPtr dataclass

Bases: RewritePattern

Source code in xdsl/transforms/convert_vector_to_ptr.py

@dataclass
class VectorLoadToPtr(RewritePattern):
    @op_type_rewrite_pattern
    def match_and_rewrite(self, op: vector.LoadOp, rewriter: PatternRewriter):
        assert isa(memref_type := op.base.type, memref.MemRefType)
        target_ptr = get_target_ptr(op.base, memref_type, op.indices, rewriter)
        rewriter.replace_op(op, ptr.LoadOp(target_ptr, op.result.type))

__init__() -> None

match_and_rewrite(op: vector.LoadOp, rewriter: PatternRewriter)

Source code in xdsl/transforms/convert_vector_to_ptr.py

@op_type_rewrite_pattern
def match_and_rewrite(self, op: vector.LoadOp, rewriter: PatternRewriter):
    assert isa(memref_type := op.base.type, memref.MemRefType)
    target_ptr = get_target_ptr(op.base, memref_type, op.indices, rewriter)
    rewriter.replace_op(op, ptr.LoadOp(target_ptr, op.result.type))

ConvertVectorToPtrPass dataclass

Bases: ModulePass

Source code in xdsl/transforms/convert_vector_to_ptr.py

@dataclass(frozen=True)
class ConvertVectorToPtrPass(ModulePass):
    name = "convert-vector-to-ptr"

    def apply(self, ctx: Context, op: builtin.ModuleOp) -> None:
        PatternRewriteWalker(
            GreedyRewritePatternApplier(
                [
                    VectorLoadToPtr(),
                    VectorStoreToPtr(),
                ]
            ),
            apply_recursively=False,
        ).rewrite_module(op)

name = 'convert-vector-to-ptr' class-attribute instance-attribute

__init__() -> None

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

Source code in xdsl/transforms/convert_vector_to_ptr.py

def apply(self, ctx: Context, op: builtin.ModuleOp) -> None:
    PatternRewriteWalker(
        GreedyRewritePatternApplier(
            [
                VectorLoadToPtr(),
                VectorStoreToPtr(),
            ]
        ),
        apply_recursively=False,
    ).rewrite_module(op)