Argmax

T-CPU ARGMAX implementation.

The CPU oracle selects token IDs from logits with deterministic tie behavior that GPU tiers can match during validation.

2 exports shown

struct

Params

pub const Params = struct

Inputs and outputs for one ARGMAX call.

function

pub fn run(params: Params) !void

Scan the logits vector and write the index of the largest element to `params.output`.

Ties resolve to the earlier index, matching the GPU oracle.

params: Logits vector and output index slot; the vector must contain at least one value.

`error.EmptyInput` when `params.logits.len == 0`, otherwise void.