package rune

You can search for identifiers within the package.

in-package search v0.2.0

package rune

rune
- Library rune
  - Rune
- Library rune.jit
  - Rune_jit
    
    Ir
    
    Dtype
    
    Var
    
    Set
    
    SymVar
    
    Special_index_kind
    
    Lowered
    
    Dtype
    
    Var
    
    Backend_intf
    
    S
    
    Device_info
    
    Renderer
    
    Compiler
    
    Runtime
    
    Debug
- Library rune.metal
  - Rune_metal
    
    Jit_backend
    
    Device_info
    
    Renderer
    
    Compiler
    
    Runtime
- Sources
  - rune
    
    autodiff.ml
    
    debug.ml
    
    jit.ml
    
    jit_xla.ml
    
    nx_rune.ml
    
    rune.ml
    
    rune__.ml
    
    tensor.ml
    
    tensor_with_debug.ml
  - rune.jit
    
    backend_intf.ml
    
    ir.ml
    
    lowerer.ml
    
    rune_jit.ml
    
    rune_jit__.ml
    
    scheduler.ml
  - rune.metal
    
    rune_metal.ml

Legend:
Page
Library
Module
Module type
Parameter
Class
Class type
Source

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;

}

package rune

Module Ir.LoweredSource

Module `Ir.Lowered`Source