matrix_8h_source.html

#pragma once


#include <mim/world.h>


#include "mim/plug/buffer/buffer.h"

#include "mim/plug/matrix/autogen.h"


namespace mim::plug::matrix {


/// Reads element `idx` from a matrix (represented by a `%buffer.Buf`), threading `mem`.


inline const Def* op_read(const Def* mem, const Def* matrix, const Def* idx) {

    auto buf_ty = Axm::isa<buffer::Buf>(matrix->type());

    if (!buf_ty) return matrix;

    auto [n, S, T] = buf_ty->args<3>();

    return buffer::op_read(n, S, T, mem, matrix, idx);

}


} // namespace mim::plug::matrix


buffer.h

mim::Axm::isa
static auto isa(const Def *def)
Definition axm.h:107

mim::Def
Base class for all Defs.
Definition def.h:261

autogen.h

mim::plug::buffer::op_read
const Def * op_read(const Def *r, const Def *s, const Def *T, const Def *mem, const Def *buf, const Def *idx)
buffer.read (r, s, T) (mem, buf, idx) ↦ [mem.M 0, T].
Definition buffer.h:22

mim::plug::matrix
The matrix Plugin
Definition matrix.h:8

mim::plug::matrix::op_read
const Def * op_read(const Def *mem, const Def *matrix, const Def *idx)
Reads element idx from a matrix (represented by a buffer.Buf), threading mem.
Definition matrix.h:11

mim::plug::mem
The mem Plugin
Definition mem.h:11

world.h