Class BufferLoad

python/tvm/tirx/expr.py:1120–1155

Buffer load node. Parameters ---------- buffer : Buffer The buffer to be loaded. indices : List[PrimExpr] The buffer indices to load values from. span : Optional[Span] The location of this expression in the source code. predicate : Optional[PrimExp

@tvm_ffi.register_object("tirx.BufferLoad")
class BufferLoad(PrimExprWithOp):
    """Buffer load node.

    Parameters
    ----------
    buffer : Buffer
        The buffer to be loaded.

    indices : List[PrimExpr]
        The buffer indices to load values from.

    span : Optional[Span]
        The location of this expression in the source code.

    predicate : Optional[PrimExpr]
        A vector mask of boolean values indicating which lanes of a vector are to be
        loaded. The number lanes of the mask must be equal to the number of lanes being loaded.
    """

    buffer: Buffer
    indices: list[PrimExpr]

    def __init__(
        self,
        buffer: Buffer,
        indices: list[PrimExpr],
        predicate: PrimExpr | None = None,
        span: Span | None = None,
    ) -> None:
        self.__init_handle_by_constructor__(
            _ffi_api.BufferLoad,
            buffer,
            indices,
            predicate,
            span,  # type: ignore
        )


@tvm_ffi.register_object("tirx.ProducerLoad")