Examples

Programs the compiler already accepts

These .kina files live in tests/compiler/examples/. Where a golden exists, the example page also shows the textual MLIR the OCaml compiler emits — the same pattern as odoc worked examples, but for the language itself.

ad_basic.kina

Automatic differentiation · MLIR golden

ad_mlp.kina

Automatic differentiation

ad_stdlib_test.kina

Automatic differentiation

add_2d.kina

Compiler example · MLIR golden

bitwise_and_i32.kina

Compiler example · MLIR golden

broadcast_row.kina

Compiler example · MLIR golden

cast_bool_i32.kina

Dtype cast · MLIR golden

cast_f16_bf16.kina

Dtype cast · MLIR golden

cast_i32.kina

Dtype cast · MLIR golden

cast_i8_i32.kina

Dtype cast · MLIR golden

concat_axis0.kina

Compiler example · MLIR golden

div_2d.kina

Compiler example · MLIR golden

effects_test.kina

Compiler example

eq_f32.kina

Compiler example · MLIR golden

frem_f32.kina

Compiler example · MLIR golden

frem_f8e4m3.kina

Compiler example · MLIR golden

gather_2d.kina

Compiler example

if_else.kina

Compiler example · MLIR golden

invalid_add_shape_mismatch.kina

Rejected by the type checker

invalid_dim_zero.kina

Rejected by the type checker

invalid_mul_shape_mismatch.kina

Rejected by the type checker

invalid_requires_mismatch.kina

Rejected by the type checker

invalid_reshape_volume_mismatch.kina

Rejected by the type checker

llm_generate.kina

Compiler example

matmul_2d.kina

Matrix multiply · MLIR golden

matmul_batch.kina

Matrix multiply · MLIR golden

max_f8e4m3.kina

Compiler example · MLIR golden

max_i32.kina

Compiler example · MLIR golden

min_f32.kina

Compiler example · MLIR golden

mlp_forward.kina

MLP forward

mlp_forward_gpu.kina

MLP forward

mlp_forward_static.kina

MLP forward

mul_2d.kina

Compiler example · MLIR golden

pow_f32.kina

Compiler example · MLIR golden

pow_f8e4m3.kina

Compiler example · MLIR golden

reduce_keep_mean_axis1.kina

Reduction · MLIR golden

reduce_max_axis1.kina

Reduction · MLIR golden

reduce_max_f8_axis0.kina

Reduction · MLIR golden

reduce_mean_f8_axis0.kina

Reduction · MLIR golden

reduce_min_axis0.kina

Reduction · MLIR golden

reduce_sum_axis1.kina

Reduction · MLIR golden

rem_i32.kina

Compiler example · MLIR golden

repeat_interleave_axis0.kina

Compiler example

reshape_2d_to_3d.kina

Compiler example · MLIR golden

scalar_bool.kina

Compiler example · MLIR golden

slice_axis0.kina

Compiler example · MLIR golden

softmax_axis1.kina

Compiler example

state_effect.kina

Compiler example

static_2d.kina

Compiler example · MLIR golden

storylm0_forward.kina

Tiny causal attention forward graph · MLIR golden

sub_2d.kina

Compiler example · MLIR golden

symbolic_3d.kina

Compiler example · MLIR golden

symbolic_requires.kina

Compiler example

transformer_ad.kina

Transformer block / AD · MLIR golden

transformer_block.kina

Transformer block / AD

transpose_2d.kina

Compiler example · MLIR golden

unary_bitwise_not_i32.kina

Unary op · MLIR golden

unary_exp_f8e4m3.kina

Unary op · MLIR golden

unary_gelu_f32.kina

Unary op · MLIR golden

unary_log_f32.kina

Unary op · MLIR golden

unary_neg_bool.kina

Unary op · MLIR golden

unary_neg_i32.kina

Unary op · MLIR golden

unary_relu.kina

Unary op · MLIR golden

unary_relu_f8e4m3.kina

Unary op · MLIR golden

unary_rsqrt_f32.kina

Unary op · MLIR golden

unary_sigmoid_f32.kina

Unary op · MLIR golden

unary_sigmoid_f8e4m3.kina

Unary op · MLIR golden

unary_silu_f32.kina

Unary op · MLIR golden

unary_sqrt_f32.kina

Unary op · MLIR golden

unary_tanh_f32.kina

Unary op · MLIR golden

while_checkpoint.kina

Compiler example · MLIR golden

while_loop.kina

Compiler example · MLIR golden