whn09 c826e81b71 neuron: mHC collapse through torch by default (DEEPINFER_NEURON_HC_TORCH=collapse); fix Neuron bind_cpu core index
collapse in torch is bit-exact with the hc_collapse kernel and a faster graph (512 rows: 0.15 vs 1.73 ms
standalone). post_torch (the kernel's operation order) is bit-equal standalone but rounds differently fused
in the layer graph, so it is opt-in. Tiny golden 9/9 with collapse. bind_cpu receives the physical core id;
_neuron_bind_cpus indexed CUDA_VISIBLE_DEVICES with it again (broke runs on cores other than 0..n).

Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
2026-10-27 09:18:56 +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: drafter forward as one region, compiled dflash metadata fill; verification + drafts regions on by default 2026-10-27 00:14:32 +08:00
models neuron: mHC collapse through torch by default (DEEPINFER_NEURON_HC_TORCH=collapse); fix Neuron bind_cpu core index 2026-10-27 09:18:56 +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