whn09 afd49fe8ce Neuron DSpark: pad multi-row decode to a power of two before the attention path; 6-row steps take the table path
The table path at 6 rows computed wrong attention (bench21: acceptance 1.27);
padded to 8 rows it is right and the step drops from ~1.8 s to ~1.07 s.
bench23 (512 in, 128 out, TP=64, nextn=5): TPOT 466.4 -> 277.4 ms, 3.85 tokens
a step, per-position acceptance identical to bench18.

Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
2026-10-24 22:31:05 +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: CP group as a replicated-decode group; MoE all-reduce mode (#28) 2026-10-23 22:46:04 +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 DSpark (#15): draft layers and main-cache writes as region graphs, decode rows padded to a power of two 2026-10-24 13:41:17 +08:00
neuron_ops Neuron DSpark: pad multi-row decode to a power of two before the attention path; 6-row steps take the table path 2026-10-24 22:31:05 +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 DSpark (#15): greedy verification and MTP ops, CP followers run the drafts 2026-10-24 11:31:59 +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: whole-layer region graphs by default (TPOT 447.8 -> 430.3 ms at TP=64) 2026-10-24 10:15:43 +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