whn09 e7cb740b97 Neuron attention: decode fp4 KV records in float arithmetic (Vector engine), not integer bit ops (GpSimd)
The compressed-KV scopes are fp4 records; their decode was nibble masks and
shifts plus the integer e2m1 / e4m3 decoders on GpSimd (three such chains were
the largest cost of the ratio-2 attention graph, device profile bench72). Now:
nibbles from floor(b / 16), e2m1 from a piecewise-linear map of the magnitude
code, scales through the hardware fp8 conversion. Bit-identical to the integer
decode for every byte (host check). bench73 (vllm-sonnet 512, TP=64, nextn=5):
TPOT 243.5 -> 228.6 ms, TTFT 4.84 -> 4.35 s, step 699 -> 666 ms, tokens per
step identical.

Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
2026-10-26 08:18:39 +08:00
..
_C initial commit 2026-10-20 12:09:42 +08:00
distributed initial commit 2026-10-20 12:09:42 +08:00
engine Neuron: sampling flags read once a step; runtime profiling on the driver only; one-lookup embedding 2026-10-25 03:52:47 +08:00
entrypoints initial commit 2026-10-20 12:09:42 +08:00
hugging_face Neuron: concatenate FP8 checkpoint tensors as bytes (fixes MoE on device) 2026-10-21 11:46:58 +08:00
model_executor Neuron: eager embedding/gather without dtype casts or host scalar operands 2026-10-25 06:55:43 +08:00
neuron_ops Neuron attention: decode fp4 KV records in float arithmetic (Vector engine), not integer bit ops (GpSimd) 2026-10-26 08:18:39 +08:00
npu_ops initial commit 2026-10-20 12:09:42 +08:00
structure_output initial commit 2026-10-20 12:09:42 +08:00
ve initial commit 2026-10-20 12:09:42 +08:00
worker Neuron table attention: decode only the blocks a decode step's requests reference 2026-10-25 17:58:11 +08:00
__init__.py Neuron basic mode: try prefill on every step 2026-10-21 21:00:54 +08:00
attrdict.py initial commit 2026-10-20 12:09:42 +08:00
config.py initial commit 2026-10-20 12:09:42 +08:00
environ.py Neuron DSpark: draft sampling as one region graph for greedy rows; token embedding on device by default 2026-10-25 06:06:50 +08:00
load_class.py initial commit 2026-10-20 12:09:42 +08:00
logger.py initial commit 2026-10-20 12:09:42 +08:00
model_config_loader.py initial commit 2026-10-20 12:09:42 +08:00
multimodal_tokens.py initial commit 2026-10-20 12:09:42 +08:00
platform.py Neuron: static decode graphs, static-shape MoE, trn2 e4m3 fp8 (issues #9, #12, #24) 2026-10-22 11:59:37 +08:00
py.typed initial commit 2026-10-20 12:09:42 +08:00
sampling_params.py initial commit 2026-10-20 12:09:42 +08:00
sequence.py initial commit 2026-10-20 12:09:42 +08:00
shared_memory.py initial commit 2026-10-20 12:09:42 +08:00
testing.py initial commit 2026-10-20 12:09:42 +08:00
timer.py initial commit 2026-10-20 12:09:42 +08:00
utils.py Neuron basic mode — part 3: end-to-end check vs the official reference 2026-10-20 20:20:02 +08:00