matrix.h

Go to the documentation of this file.

#pragma once
 
#include <mim/pass/pipelinebuilder.h>
#include <mim/world.h>
 
#include "mim/plug/matrix/autogen.h"
 
namespace mim::plug::matrix {
 
#define INTERNAL_PREFIX "internal_mapRed_"
 
/// %mat.zero: Π [n: .Nat, S: «n; .Nat», m: .Nat] -> %mat.Mat (n,S,(.Idx m));
inline const Def* zero_int(World& w, Ref n, Ref S, Ref mem, nat_t m) {
    // TODO: use mim definition by name
    return w.app(w.annex<matrix::constMat>(), {n, S, w.type_idx(m), mem, w.lit_idx(m, 0)});
}
 
inline const Def* op_read(Ref mem, Ref matrix, Ref idx) {
    auto& world = matrix->world();
    auto mat_ty = match<Mat>(matrix->type());
    if (!mat_ty) return matrix;
    assert(mat_ty);
    world.DLOG("matrix read: {}[{}]", matrix, idx);
    world.DLOG(" matrix type: {}", matrix->type());
    auto [n, S, T] = mat_ty->args<3>();
    world.DLOG(" (n,S,T): {}, {}, {}", n, S, T);
    return world.app(world.app(world.annex<read>(), {n, S, T}), {mem, matrix, idx});
}
 
} // namespace mim::plug::matrix