Module Sarek_metal.Sarek_ir_metal

Re-export the Metal code generator from the pure sarek_codegen library. Consumers of Sarek_metal.Sarek_ir_metal and in-package Sarek_ir_metal are unchanged.

include module type of struct include Sarek_codegen.Sarek_ir_metal end

Local error module — same raised exception as the package-level Metal_error.

val current_framework : string option Stdlib.ref

Current framework string for SNative code generation. Always None in normal use; SNative branches check this ref and error if None.

val current_variants : (string * (string * Sarek_ir_types.elttype list) list) list Stdlib.ref

Current kernel's variant definitions (set during generate)

Type Mapping

val mangle_name : string -> string
val metal_type_of_elttype : Sarek_ir_types.elttype -> string

Map Sarek IR element type to Metal C type string

val metal_memspace : Sarek_ir_types.memspace -> string

Map memory space to Metal qualifier

val metal_param_type : Sarek_ir_types.elttype -> string

Map Sarek IR element type to Metal C type for kernel parameters

val metal_helper_param_type : Sarek_ir_types.elttype -> string

Map Sarek IR element type to Metal C type for helper function parameters

val metal_atomic_type_of_elttype : Sarek_ir_types.elttype -> string

Convert type to atomic type for Metal

Thread Intrinsics

val metal_thread_intrinsic : string -> string

Expression Generation

val gen_expr : Stdlib.Buffer.t -> Sarek_ir_types.expr -> unit
val gen_binop : Sarek_ir_types.binop -> string
val gen_unop : Sarek_ir_types.unop -> string
val gen_intrinsic : Stdlib.Buffer.t -> string list -> string -> Sarek_ir_types.expr list -> unit

L-value Generation

val gen_lvalue : Stdlib.Buffer.t -> Sarek_ir_types.lvalue -> unit

Statement Generation

val indent_nested : string -> string

Nested indentation level

val gen_match_pattern : Stdlib.Buffer.t -> string -> string -> string -> string list -> (string -> Sarek_ir_types.elttype list option) -> unit

Generate match case pattern with variable bindings

val gen_var_decl : Stdlib.Buffer.t -> string -> string -> Sarek_ir_types.elttype -> Sarek_ir_types.expr -> unit

Generate variable declaration with initialization

val gen_array_decl : Stdlib.Buffer.t -> string -> string -> Sarek_ir_types.elttype -> Sarek_ir_types.expr -> string -> unit

Generate array declaration

val gen_stmt : Stdlib.Buffer.t -> string -> Sarek_ir_types.stmt -> unit

Declaration Generation

val is_vec_type : Sarek_ir_types.elttype -> bool

Check if a type is a vector (requires length parameter)

val gen_param_metal : Stdlib.Buffer.t -> string list -> int -> Sarek_ir_types.decl -> int

Generate parameter with Metal buffer attributes, returns next buffer index

val gen_param : Stdlib.Buffer.t -> Sarek_ir_types.decl -> unit
val collect_atomic_vars_expr : Sarek_ir_types.expr -> string list

Collect variable names used in atomic operations

val collect_atomic_vars_lvalue : Sarek_ir_types.lvalue -> string list
val collect_atomic_vars_stmt : Sarek_ir_types.stmt -> string list
val gen_local : Stdlib.Buffer.t -> string -> string list -> Sarek_ir_types.decl -> unit

Helper Function Generation

val gen_helper_func : Stdlib.Buffer.t -> Sarek_ir_types.helper_func -> unit

Generate a helper function (Metal device function)

Kernel Generation

val pretty_print_metal : string -> string

Pretty-print Metal source code

val generate : Sarek_ir_types.kernel -> string

Generate complete Metal source for a kernel

val gen_variant_def : Stdlib.Buffer.t -> (string * (string * Sarek_ir_types.elttype list) list) -> unit

Generate variant type definition for Metal

val generate_with_types : types:(string * (string * Sarek_ir_types.elttype) list) list -> Sarek_ir_types.kernel -> string

Generate Metal source with custom type definitions

val generate_with_fp64 : Sarek_ir_types.kernel -> string

Generate Metal source with double precision (Metal supports FP64 natively)