metaltile-std 0.1.0

MetalTile kernel standard library — benchmark metadata and type definitions
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26

//! Copyright 2026 0xClandestine, Ekryski, TheTom, Ambisphaeric
//! SPDX-License-Identifier: Apache-2.0
//! Random benchmark — #[kernel] DSL vs MLX metal/random.metal

use metaltile::{bench_kernel, kernel};

#[bench_kernel(
    op="random",
    subop="random_hash",
    class=Random,
    n=1048576,
    tpg=1024,
    tol=0.0,
    mlx="rbitsc",
    metal_file="random.metal",
    dtypes=crate::spec::F32_ONLY,
)]
#[kernel]
pub fn mt_random_hash(out: Tensor<u32>, #[constexpr] n: u32) {
    let gid = program_id::<0>();
    let mut s = gid + 1u32;
    s = s ^ (s << 13u32);
    s = s ^ (s >> 17u32);
    s = s ^ (s << 5u32);
    store(out[gid], s);
}