1/* @ajax/mesh-ref-gpt2-logits: what is done with GPT-2's logits (mesh-ref gpt2.rs `argmax`,
2 * `digest_extend`): the greedy pick and one digest over every step of a generation. Freestanding. */
3#ifndef MESH_REF_GPT2_LOGITS_H
4#define MESH_REF_GPT2_LOGITS_H
5
6#include <stddef.h>
7#include <stdint.h>
8
9#define MESH_GPT2_DIGEST_SEED 0xcbf29ce484222325ull
10
11/* The first index of the largest logit (FTZ compare). */
12uint32_t mesh_gpt2_argmax(const float *l, size_t n);
13/* FNV-1a 64 over the logits' bit words, continued from h (start: MESH_GPT2_DIGEST_SEED). */
14uint64_t mesh_gpt2_digest_extend(uint64_t h, const float *l, size_t n);
15
16#endif