whn09 cedd237772 neuron: fp8-activation linears take the NKI kernel (Engram projection); drop unused fp8 hidden copies before region splits; log weights without NKI scales ... An fp8 tuple input fell back to the deep_gemm shim, which dequantized the whole fp8 weight in the graph on every call (Engram fused_kv_projection: 9 ms of an Engram layer's prefill graph, GpSimd POW for the scales). Now decoded to bf16 and run through the NKI kernel. TTFT 1.71 -> 1.66 s (bucket 512), tokens unchanged. Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>		2026-10-27 11:50:34 +08:00
..
eplb	initial commit	2026-10-20 12:09:42 +08:00
input_metadata	initial commit	2026-10-20 12:09:42 +08:00
layers	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
models	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
parallel_utils	initial commit	2026-10-20 12:09:42 +08:00
__init__.py	initial commit	2026-10-20 12:09:42 +08:00
layer_utils.py	initial commit	2026-10-20 12:09:42 +08:00
model_loader.py	neuron: shared expert sliced over ranks, summed in the MoE all-reduce (DEEPINFER_NEURON_SHARED_SLICE=1)	2026-10-27 01:31:41 +08:00
model_metadata_utils.py	Neuron: prompt chunks through the decode region graphs (issue #26)	2026-10-23 03:19:56 +08:00
model_output.py	initial commit	2026-10-20 12:09:42 +08:00
sampling_inputs.py	initial commit	2026-10-20 12:09:42 +08:00
utils.py	Neuron: static decode graphs, static-shape MoE, trn2 e4m3 fp8 (issues #9, #12, #24)	2026-10-22 11:59:37 +08:00