package rune

You can search for identifiers within the package.

in-package search v0.2.0

Automatic differentiation and JIT compilation for OCaml

Install

Dune Dependency

github.com Readme Changelog Edit opam file Versions (1)

Authors

Thibaut Mattio

Maintainers

Thibaut Mattio

Sources

raven-1.0.0.alpha0.tbz

sha256=a9a8a9787f8250337187bb7b21cb317c41bfd2ecf08bcfe0ab407c7b6660764d

sha512=fe13cf257c487e41efe2967be147d80fa94bac8996d3aab2b8fd16f0bbbd108c15e0e58c025ec9bf294d4a0d220ca2ba00c3b1b42fa2143f758c5f0ee4c15782

doc/rune.jit/Rune_jit/Ir/Lowered/index.html

Module `Ir.Lowered`Source

Sourcetype alu_op =

| Binary of binop_kind
| Unary of unary_op_kind
| Ternary of ternary_op_kind

Sourcetype instruction =

| L_Buffer of {
1. dtype : Dtype.any;
2. size : int;
3. out : Var.t;
}
| L_Local of {
1. dtype : Dtype.any;
2. size : int;
3. out : Var.t;
}
| L_Acc of {
1. dtype : Dtype.any;
2. out : Var.t;
}
| L_Define_Global of {
1. ptr : Var.t;
2. dtype : Dtype.any;
3. size : int;
}
| L_Const of {
1. dtype : Dtype.any;
2. value : string;
3. out : Var.t;
}
| L_Vconst of {
1. dst : Var.t;
2. values : string array;
3. dtype : Dtype.any;
}
| L_Special of {
1. dst : Var.t;
2. kind : Special_index_kind.t;
}
| L_Define_Var of {
1. sym_var : SymVar.t;
2. out : Var.t;
}
| L_Range of {
1. idx : Var.t;
2. bound : Var.t;
}
| L_EndRange
| L_If of {
1. cond : Var.t;
}
| L_EndIf
| L_Barrier
| L_Block of {
1. block_id : int;
2. start : bool;
}
| L_Unroll of {
1. idx : Var.t;
2. iterations : int;
}
| L_Load of {
1. dst : Var.t;
2. buf : Var.t;
3. idx : Var.t;
4. dtype : Dtype.any;
5. valid : Var.t option;
}
| L_Store of {
1. buf : Var.t;
2. idx : Var.t;
3. src : Var.t;
4. valid : Var.t option;
}
| L_ALU of {
1. dst : Var.t;
2. op : alu_op;
3. args : Var.t list;
4. dtype : Dtype.any;
}
| L_Gep of {
1. dst : Var.t;
2. src : Var.t;
3. indices : int array;
4. dtype : Dtype.any;
}
| L_Vectorize of {
1. dst : Var.t;
2. srcs : Var.t array;
3. dtype : Dtype.any;
}
| L_Ptrcat of {
1. dst : Var.t;
2. ptrs : Var.t array;
3. dtype : Dtype.any;
}
| L_Wmma of {
1. dst : Var.t;
2. a : Var.t;
3. b : Var.t;
4. c : Var.t;
5. m : int;
6. n : int;
7. k : int;
8. dtype : Dtype.any;
}
| L_Cast of {
1. dst : Var.t;
2. src : Var.t;
3. dtype : Dtype.any;
}
| L_Bitcast of {
1. dst : Var.t;
2. src : Var.t;
3. dtype : Dtype.any;
}
| L_Assign of {
1. dst : Var.t;
2. src : Var.t;
}
| L_Custom of {
1. dst : Var.t option;
2. op_name : string;
3. args : Var.t array;
4. attributes : (string * custom_attr) list;
5. inline : bool;
}
| L_Noop

Sourcetype graph_t = {

instructions : instruction list;
vars_metadata : (Var.t, var_metadata) Hashtbl.t;
kernel_input_vars : Var.t list;
kernel_output_vars : Var.t list;
symbolic_vars : SymVar.t list;

}