tensorlogic-compiler 0.1.0-rc.1

Compiler for transforming logic expressions into tensor computation graphs
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25

//! Predicate compilation.

use anyhow::Result;
use tensorlogic_ir::{EinsumGraph, Term};

use crate::context::{CompileState, CompilerContext};

pub(crate) fn compile_predicate(
    name: &str,
    args: &[Term],
    ctx: &mut CompilerContext,
    graph: &mut EinsumGraph,
) -> Result<CompileState> {
    for arg in args {
        if let Term::Var(v) = arg {
            ctx.assign_axis(v);
        }
    }

    let axes = ctx.get_axes(args)?;
    let tensor_name = format!("{}[{}]", name, axes);
    let tensor_idx = graph.add_tensor(tensor_name);

    Ok(CompileState { tensor_idx, axes })
}