skip to content
==> program.toml <==
1# gpt2x: GPT-2 124M's exact-mode kernels on the GPU (CUDA and Metal), the
2# exact path of programs/gpt2 (its worker loads this image in-process through
3# mesh-host's backend: mesh-host gpt2x.rs is the host side).  Every op follows
4# mesh_ref::gpt2 operation for operation (FTZ, correctly rounded, unfused,
5# fixed sum orders; exp/tanh/sqrt from + - * /), so the logits and the KV
6# cache are bitwise the reference's on every GPU (kcheck GX*).  8-warp nodes;
7# the weights (f32, 498 MB), the KV cache (12 x 1024 x 768 x 2 f32) and the
8# activations of up to 1024 rows live in the scratch.  Ops and the view:
9# mesh_abi::v2::views::gpt2x.
10name          = "gpt2x"
11abi           = 6
12view_bytes    = 0x700000
13nodes         = "fit"
14warps         = 8
15shared_bytes  = 0x4200
16scratch_bytes = 0x24200000
17policy        = "on_demand"
18fanout        = 32
19counters      = ["tasks", "rows", "tiles", "chunks"]
20fields        = ["magic:u32:0x0"]
21backends      = ["cuda", "metal"]