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 (2)

Authors

Thibaut Mattio

Maintainers

Thibaut Mattio

Sources

raven-1.0.0.alpha1.tbz

sha256=8e277ed56615d388bc69c4333e43d1acd112b5f2d5d352e2453aef223ff59867

sha512=369eda6df6b84b08f92c8957954d107058fb8d3d8374082e074b56f3a139351b3ae6e3a99f2d4a4a2930dd950fd609593467e502368a13ad6217b571382da28c

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