Floats

1import { Cursor, advance } from "./core.js";
2
3/**
4 * Scratch view for widening a `BFloat16`: its 16 bits are the top half of an
5 * IEEE 754 float32, so we stage them into a 4-byte buffer and read a float32.
6 */
7const bf16Scratch = new DataView(new ArrayBuffer(4));
8
9/** Read a `Float32`: 4 bytes, little-endian IEEE 754 single precision. */
10export function readFloat32(state: Cursor): number {
11  return state.view.getFloat32(advance(state, 4), true);
12}
13
14/** Read a `Float64`: 8 bytes, little-endian IEEE 754 double precision. */
15export function readFloat64(state: Cursor): number {
16  return state.view.getFloat64(advance(state, 8), true);
17}
18
19/**
20 * Read a `BFloat16`: 2 bytes, little-endian. BFloat16 is the high 16 bits of a
21 * float32 (same 8-bit exponent, 7-bit mantissa), so placing the bits in the top
22 * half of a 32-bit float and reading it back is exact.
23 *
24 * NOTE: `bf16Scratch` is module-level shared state written-then-read in this
25 * function. That is safe because the read is synchronous; do NOT introduce an
26 * `await`/`yield` between the `setUint32` and the `getFloat32`.
27 */
28export function readBFloat16(state: Cursor): number {
29  const bits = state.view.getUint16(advance(state, 2), true);
30  bf16Scratch.setUint32(0, bits * 0x10000, true);
31  return bf16Scratch.getFloat32(0, true);
32}