sarek 20210823 · OCaml Package

type customtypes =

| KRecord of Camlp4.PreCast.Syntax.Ast.ctyp list * Camlp4.PreCast.Syntax.Ast.ident list * bool list
| KSum of (string * Camlp4.PreCast.Syntax.Ast.ctyp option) list

type memspace =

| Local
| Global
| Shared
| Any

type ktyp =

| TUnknown
| TUnit
| TInt32
| TInt64
| TFloat32
| TFloat64
| TBool
| TAny
| TVec of ktyp
| TArr of ktyp * memspace
| TApp of ktyp * ktyp
| Custom of customtypes * string

val ktyp_to_string : ktyp -> string

val ex32 : Camlp4.PreCast.Syntax.Ast.expr

val ex64 : Camlp4.PreCast.Syntax.Ast.expr

val extensions : Camlp4.PreCast.Syntax.Ast.expr list ref

type var = {

n : int;
mutable var_type : ktyp;
mutable is_mutable : bool;
mutable read_only : bool;
mutable write_only : bool;
mutable is_global : bool;

}

val new_kernel : bool ref

exception ArgumentError

exception Unbound_value of string * Camlp4.PreCast.Syntax.Ast.Loc.t

exception Unbound_module of string * Camlp4.PreCast.Syntax.Ast.Loc.t

exception Immutable of string * Camlp4.PreCast.Syntax.Ast.Loc.t

exception TypeError of ktyp * ktyp * Camlp4.PreCast.Syntax.Ast.Loc.t

exception FieldError of string * string * Camlp4.PreCast.Syntax.Ast.Loc.t

type cfun = {

nb_args : int;
cuda_val : string;
opencl_val : string;
typ : ktyp;

}

type k_expr =

| Open of Camlp4.PreCast.Syntax.Ast.Loc.t * Camlp4.PreCast.Syntax.Ast.ident * kexpr
| App of Camlp4.PreCast.Syntax.Ast.Loc.t * kexpr * kexpr list
| Acc of Camlp4.PreCast.Syntax.Ast.Loc.t * kexpr * kexpr
| VecSet of Camlp4.PreCast.Syntax.Ast.Loc.t * kexpr * kexpr
| VecGet of Camlp4.PreCast.Syntax.Ast.Loc.t * kexpr * kexpr
| ArrSet of Camlp4.PreCast.Syntax.Ast.Loc.t * kexpr * kexpr
| ArrGet of Camlp4.PreCast.Syntax.Ast.Loc.t * kexpr * kexpr
| Seq of Camlp4.PreCast.Syntax.Ast.Loc.t * kexpr * kexpr
| Fun of Camlp4.PreCast.Syntax.Ast.Loc.t * Camlp4.PreCast.Syntax.Ast.expr * ktyp * cfun * string list
| Bind of Camlp4.PreCast.Syntax.Ast.Loc.t * kexpr * kexpr * kexpr * bool
| Plus32 of Camlp4.PreCast.Syntax.Ast.Loc.t * kexpr * kexpr
| Plus64 of Camlp4.PreCast.Syntax.Ast.Loc.t * kexpr * kexpr
| PlusF32 of Camlp4.PreCast.Syntax.Ast.Loc.t * kexpr * kexpr
| PlusF64 of Camlp4.PreCast.Syntax.Ast.Loc.t * kexpr * kexpr
| Record of Camlp4.PreCast.Syntax.Ast.Loc.t * field list
| RecGet of Camlp4.PreCast.Syntax.Ast.Loc.t * kexpr * Camlp4.PreCast.Syntax.Ast.ident
| RecSet of Camlp4.PreCast.Syntax.Ast.Loc.t * kexpr * kexpr
| Map of Camlp4.PreCast.Syntax.Ast.Loc.t * kexpr * kexpr * kexpr
| Min32 of Camlp4.PreCast.Syntax.Ast.Loc.t * kexpr * kexpr
| Min64 of Camlp4.PreCast.Syntax.Ast.Loc.t * kexpr * kexpr
| MinF32 of Camlp4.PreCast.Syntax.Ast.Loc.t * kexpr * kexpr
| MinF64 of Camlp4.PreCast.Syntax.Ast.Loc.t * kexpr * kexpr
| Mul32 of Camlp4.PreCast.Syntax.Ast.Loc.t * kexpr * kexpr
| Mul64 of Camlp4.PreCast.Syntax.Ast.Loc.t * kexpr * kexpr
| MulF32 of Camlp4.PreCast.Syntax.Ast.Loc.t * kexpr * kexpr
| MulF64 of Camlp4.PreCast.Syntax.Ast.Loc.t * kexpr * kexpr
| Mod of Camlp4.PreCast.Syntax.Ast.Loc.t * kexpr * kexpr
| Div32 of Camlp4.PreCast.Syntax.Ast.Loc.t * kexpr * kexpr
| Div64 of Camlp4.PreCast.Syntax.Ast.Loc.t * kexpr * kexpr
| DivF32 of Camlp4.PreCast.Syntax.Ast.Loc.t * kexpr * kexpr
| DivF64 of Camlp4.PreCast.Syntax.Ast.Loc.t * kexpr * kexpr
| Id of Camlp4.PreCast.Syntax.Ast.Loc.t * Camlp4.PreCast.Syntax.Ast.ident
| Int of Camlp4.PreCast.Syntax.Ast.Loc.t * string
| Int32 of Camlp4.PreCast.Syntax.Ast.Loc.t * string
| Int64 of Camlp4.PreCast.Syntax.Ast.Loc.t * string
| Float of Camlp4.PreCast.Syntax.Ast.Loc.t * string
| Float32 of Camlp4.PreCast.Syntax.Ast.Loc.t * string
| Float64 of Camlp4.PreCast.Syntax.Ast.Loc.t * string
| BoolNot of Camlp4.PreCast.Syntax.Ast.Loc.t * kexpr
| BoolAnd of Camlp4.PreCast.Syntax.Ast.Loc.t * kexpr * kexpr
| BoolOr of Camlp4.PreCast.Syntax.Ast.Loc.t * kexpr * kexpr
| BoolEq of Camlp4.PreCast.Syntax.Ast.Loc.t * kexpr * kexpr
| BoolEq32 of Camlp4.PreCast.Syntax.Ast.Loc.t * kexpr * kexpr
| BoolEq64 of Camlp4.PreCast.Syntax.Ast.Loc.t * kexpr * kexpr
| BoolEqF32 of Camlp4.PreCast.Syntax.Ast.Loc.t * kexpr * kexpr
| BoolEqF64 of Camlp4.PreCast.Syntax.Ast.Loc.t * kexpr * kexpr
| BoolLt of Camlp4.PreCast.Syntax.Ast.Loc.t * kexpr * kexpr
| BoolLt32 of Camlp4.PreCast.Syntax.Ast.Loc.t * kexpr * kexpr
| BoolLt64 of Camlp4.PreCast.Syntax.Ast.Loc.t * kexpr * kexpr
| BoolLtF32 of Camlp4.PreCast.Syntax.Ast.Loc.t * kexpr * kexpr
| BoolLtF64 of Camlp4.PreCast.Syntax.Ast.Loc.t * kexpr * kexpr
| BoolLtE of Camlp4.PreCast.Syntax.Ast.Loc.t * kexpr * kexpr
| BoolLtE32 of Camlp4.PreCast.Syntax.Ast.Loc.t * kexpr * kexpr
| BoolLtE64 of Camlp4.PreCast.Syntax.Ast.Loc.t * kexpr * kexpr
| BoolLtEF32 of Camlp4.PreCast.Syntax.Ast.Loc.t * kexpr * kexpr
| BoolLtEF64 of Camlp4.PreCast.Syntax.Ast.Loc.t * kexpr * kexpr
| BoolGt of Camlp4.PreCast.Syntax.Ast.Loc.t * kexpr * kexpr
| BoolGt32 of Camlp4.PreCast.Syntax.Ast.Loc.t * kexpr * kexpr
| BoolGt64 of Camlp4.PreCast.Syntax.Ast.Loc.t * kexpr * kexpr
| BoolGtF32 of Camlp4.PreCast.Syntax.Ast.Loc.t * kexpr * kexpr
| BoolGtF64 of Camlp4.PreCast.Syntax.Ast.Loc.t * kexpr * kexpr
| BoolGtE of Camlp4.PreCast.Syntax.Ast.Loc.t * kexpr * kexpr
| BoolGtE32 of Camlp4.PreCast.Syntax.Ast.Loc.t * kexpr * kexpr
| BoolGtE64 of Camlp4.PreCast.Syntax.Ast.Loc.t * kexpr * kexpr
| BoolGtEF32 of Camlp4.PreCast.Syntax.Ast.Loc.t * kexpr * kexpr
| BoolGtEF64 of Camlp4.PreCast.Syntax.Ast.Loc.t * kexpr * kexpr
| Ife of Camlp4.PreCast.Syntax.Ast.Loc.t * kexpr * kexpr * kexpr
| If of Camlp4.PreCast.Syntax.Ast.Loc.t * kexpr * kexpr
| Match of Camlp4.PreCast.Syntax.Ast.Loc.t * kexpr * case list
| DoLoop of Camlp4.PreCast.Syntax.Ast.Loc.t * kexpr * kexpr * kexpr * kexpr
| While of Camlp4.PreCast.Syntax.Ast.Loc.t * kexpr * kexpr
| End of Camlp4.PreCast.Syntax.Ast.Loc.t * kexpr
| Ref of Camlp4.PreCast.Syntax.Ast.Loc.t * kexpr
| ModuleAccess of Camlp4.PreCast.Syntax.Ast.Loc.t * string * kexpr
| True of Camlp4.PreCast.Syntax.Ast.Loc.t
| False of Camlp4.PreCast.Syntax.Ast.Loc.t
| TypeConstraint of Camlp4.PreCast.Syntax.Ast.Loc.t * kexpr * ktyp
| Nat of Camlp4.PreCast.Syntax.Ast.Loc.t * Camlp4.PreCast.Syntax.Ast.expr
| Pragma of Camlp4.PreCast.Syntax.Ast.Loc.t * string list * kexpr
| Noop

and case = Camlp4.PreCast.Syntax.Ast.Loc.t * pattern * kexpr

and field =
  Camlp4.PreCast.Syntax.Ast.Loc.t * Camlp4.PreCast.Syntax.Ast.ident * kexpr

and pattern =

| Constr of string * Camlp4.PreCast.Syntax.Ast.ident option

and kexpr = {

mutable t : ktyp;
mutable e : k_expr;
loc : Camlp4.PreCast.Syntax.Ast.Loc.t;

}

val is_unknown : ktyp -> bool

val update_type : kexpr -> ktyp -> unit

val string_of_ident : Camlp4.PreCast.Syntax.Ast.ident -> string

val k_expr_to_string : k_expr -> string

val expr_of_patt : 
  Camlp4.PreCast.Syntax.Ast.patt ->
  Camlp4.PreCast.Syntax.Ast.expr

type k_patt

type spoc_module = {

mod_name : string;
mod_constants : (ktyp * string * string * string) list;
mod_functions : (ktyp * string * int * string * string) list;
mod_modules : (string, spoc_module) Hashtbl.t;

}

val return_type : ktyp ref

val arg_idx : int ref

val args : unit -> (string, var) Hashtbl.t

type recrd_field = {

id : int;
name : string;
field : string;
mutable ctyps : string list;

}

type cstr = {

id : int;
name : string;
mutable nb_args : int;
mutable ctyp : string;
typ : customtypes;

}

type localfun = cfun * Camlp4.PreCast.Syntax.Ast.str_item * string list

val current_args : (string, var) Hashtbl.t ref

val intrinsics_fun : (string, cfun) Hashtbl.t ref

val global_fun : (string, cfun) Hashtbl.t ref

val local_fun : (string, localfun) Hashtbl.t ref

val intrinsics_const : (string, cfun) Hashtbl.t ref

val constructors : (string, cstr) Hashtbl.t ref

val rec_fields : (string, recrd_field) Hashtbl.t ref

val custom_types : (string, customtypes) Hashtbl.t

val arg_list : Camlp4.PreCast.Syntax.Ast.expr list ref

val n_lifted_vals : int ref

val ktyp_of_typ : Camlp4.PreCast.Syntax.Ast.ctyp -> ktyp

val new_arg_of_patt : Camlp4.PreCast.Syntax.Ast.patt -> unit

val std : spoc_module

val vector : spoc_module

val mathf32 : spoc_module

val mathf64 : spoc_module

val math : spoc_module

val modules : (string, spoc_module) Hashtbl.t

val open_module : string -> Camlp4.PreCast.Syntax.Ast.Loc.t -> unit

val close_module : string -> unit

val ctype_of_sarek_type : string -> string

val mltype_of_sarek_type : string -> string

val string_of_ctyp : Camlp4.PreCast.Syntax.Ast.ctyp -> string

val sarek_types_tbl : (string, string) Hashtbl.t

val get_sarek_name : string -> string

val ident_of_patt : 'a -> pattern -> int

val type_of_patt : pattern -> string

package sarek