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-project

Dependency

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

Authors

Thibaut Mattio

Maintainers

Thibaut Mattio

Sources

raven-1.0.0.alpha2.tbz

sha256=93abc49d075a1754442ccf495645bc4fdc83e4c66391ec8aca8fa15d2b4f44d2

sha512=5eb958c51f30ae46abded4c96f48d1825f79c7ce03f975f9a6237cdfed0d62c0b4a0774296694def391573d849d1f869919c49008acffca95946b818ad325f6f

doc/rune.jit/Rune_jit/Internal/Lowerer/index.html

Module `Internal.Lowerer`

type kernel_spec

val const_to_string : 'a. 'a Ir.Dtype.t -> 'a -> string

type lowering_ctx = {

meta : (Ir.Var.t, Ir.var_metadata) Hashtbl.t;
scalar_map : (Ir.Var.t, Ir.Var.t) Hashtbl.t;
buffer_map : (Ir.Var.t, Ir.Var.t) Hashtbl.t;
instrs : Ir.Lowered.instruction list ref;

}

val new_ctx : 
  (Ir.Var.t, Ir.var_metadata) Hashtbl.t ->
  (Ir.Var.t, Ir.var_metadata) Hashtbl.t ->
  lowering_ctx

val add_instr : lowering_ctx -> Ir.Lowered.instruction -> unit

val ensure_meta : lowering_ctx -> Ir.Var.t -> Ir.var_metadata -> unit

val meta_of : lowering_ctx -> Ir.Var.t -> Ir.var_metadata option

val ll_of_hl : lowering_ctx -> Ir.Var.t -> buffer:bool -> Ir.Var.t

val gtid : lowering_ctx -> int

val load_scalar : 
  lowering_ctx ->
  hl_buffer:Ir.Var.t ->
  idx:int ->
  dtype:'a Ir.Dtype.t ->
  Ir.Var.t

val lower_node : lowering_ctx -> Ir.any_node -> Ir.Var.t list -> unit

val lower_kernel : 
  kernel_spec:kernel_spec ->
  original_graph_vars_metadata:(Ir.Var.t, Ir.var_metadata) Hashtbl.t ->
  Ir.Lowered.graph_t

val lower_from_scheduled : 
  item:Ir.Scheduled.schedule_item ->
  vars_metadata:(Ir.Var.t, Ir.var_metadata) Hashtbl.t ->
  (Ir.Lowered.graph_t, string) result