whn09 7f558a3775 neuron: workers bind memory to their own core's NUMA node; prompt chunks run the attention half as one graph; glibc trims freed heap early (TTFT 796 -> 720 ms on the bench prompt, jitter +-25 -> +-3 ms)
- NUMA: each worker sees one core, so which_numa(local_rank) failed on all but local rank 0 and every worker
  inherited its MPOL_BIND to node 1: all host memory landed on one node and a run could be OOM-killed.
- DEEPINFER_NEURON_WHOLE_ATTN_MIN_ROWS=512: no cuts inside the attention half of a >=512-row step, only before
  the MoE (its collectives are another group kind).
- MALLOC_TRIM_THRESHOLD_/MALLOC_TOP_PAD_ defaults: compile-time host memory per rank stops growing ~15 GB.
  (MALLOC_ARENA_MAX=2 bounds it too but costs ~50 ms TTFT.)

Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
2026-10-29 12:10:31 +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: 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
models neuron: workers bind memory to their own core's NUMA node; prompt chunks run the attention half as one graph; glibc trims freed heap early (TTFT 796 -> 720 ms on the bench prompt, jitter +-25 -> +-3 ms) 2026-10-29 12:10:31 +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: norms feeding only NKI kernels skip their dead fp8 copy (TTFT 1387 -> 1229 ms, step 457 -> 450 ms, tokens identical); capacity MoE wired into the layer region graph with its overflow flag riding the MoE all-reduce (still off: per-layer host sync makes prefill slower, 1823 ms) 2026-10-28 05:29:40 +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