whn09 d7650b21c3 neuron: e4m3 KV-cache codes through the hardware conversion in an NKI kernel (neuron_ops/nki/fp8_encode.py; 512x512 0.63 -> 0.11 ms, tokens identical) ... Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>		2026-10-29 00:21:02 +08:00
..
tilelang_ops	neuron: e4m3 KV-cache codes through the hardware conversion in an NKI kernel (neuron_ops/nki/fp8_encode.py; 512x512 0.63 -> 0.11 ms, tokens identical)	2026-10-29 00:21:02 +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