whn09 deb8c5a958 neuron: single-request prompt chunks attend densely over their own SWA ring and compressed blocks, masks from positions (DEEPINFER_NEURON_PREFILL_DENSE, check mode DEEPINFER_NEURON_PREFILL_DENSE_CHECK) ... Within 1 bf16 ulp of the gathered kernel on the full model (mask equality checked on every layer). With a 1024 context bucket: TTFT 1190 -> 1113 ms; tokens differ from the kernel's by rounding (2.89 -> 3.02 tokens a step). Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>		2026-10-28 09:09:34 +08:00
..
tilelang_ops	neuron: drafter forward as one region, compiled dflash metadata fill; verification + drafts regions on by default	2026-10-27 00:14:32 +08:00
__init__.py	initial commit	2026-10-20 12:09:42 +08:00
attention_backend.py	neuron: single-request prompt chunks attend densely over their own SWA ring and compressed blocks, masks from positions (DEEPINFER_NEURON_PREFILL_DENSE, check mode DEEPINFER_NEURON_PREFILL_DENSE_CHECK)	2026-10-28 09:09:34 +08:00
layernorm.py	initial commit	2026-10-20 12:09:42 +08:00
linear.py	neuron: fp8-activation linears take the NKI kernel (Engram projection); drop unused fp8 hidden copies before region splits; log weights without NKI scales	2026-10-27 11:50:34 +08:00
mtp_info_pool.py	initial commit	2026-10-20 12:09:42 +08:00
rotary_embedding.py	initial commit	2026-10-20 12:09:42 +08:00
sampler_utils.py	initial commit	2026-10-20 12:09:42 +08:00
sampler.py	neuron: compiled decode verification behind DEEPINFER_NEURON_COMPILE_VERIFY (default off)	2026-10-26 21:33:34 +08:00
vocab_parallel_embedding.py	Neuron: eager embedding/gather without dtype casts or host scalar operands	2026-10-25 06:55:43 +08:00