whn09 f07d0a7c6f Neuron: compiled log-softmax + greedy argmax in the sampler; eager op census (#27) ... The op census of a decode step (DEEPINFER_NEURON_REGION_TIMING) found 85 eager aten ops in the forward outside the region graphs but 1905 in sampling (the top-p/top-k bisection and the int64 philox draw), ~0.6 s of NEFF launches. - The log-softmax / top-p / top-k part runs as one compiled graph: 1.4 ms instead of 46.7 ms on one core for the [1, 4096] force_topk row, logprobs within 1e-9. - Greedy rows (temperature clamped to SAMPLING_EPS) have one-hot weights, so the two-level draw returns their argmax; an all-greedy batch takes the argmax from the graph and skips the draw. The philox draw stays eager for sampled rows: neuronx-cc graphs reject 64-bit constants outside the int32 range (NCC_ESFH001), an int32 philox is #3. Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>		2026-10-23 09:42:08 +08:00
..
_C	initial commit	2026-10-20 12:09:42 +08:00
distributed	initial commit	2026-10-20 12:09:42 +08:00
engine	initial commit	2026-10-20 12:09:42 +08:00
entrypoints	initial commit	2026-10-20 12:09:42 +08:00
hugging_face	Neuron: concatenate FP8 checkpoint tensors as bytes (fixes MoE on device)	2026-10-21 11:46:58 +08:00
model_executor	Neuron: compiled log-softmax + greedy argmax in the sampler; eager op census (#27)	2026-10-23 09:42:08 +08:00
neuron_ops	Neuron: compiled log-softmax + greedy argmax in the sampler; eager op census (#27)	2026-10-23 09:42:08 +08:00
npu_ops	initial commit	2026-10-20 12:09:42 +08:00
structure_output	initial commit	2026-10-20 12:09:42 +08:00
ve	initial commit	2026-10-20 12:09:42 +08:00
worker	Neuron: compiled log-softmax + greedy argmax in the sampler; eager op census (#27)	2026-10-23 09:42:08 +08:00
__init__.py	Neuron basic mode: try prefill on every step	2026-10-21 21:00:54 +08:00
attrdict.py	initial commit	2026-10-20 12:09:42 +08:00
config.py	initial commit	2026-10-20 12:09:42 +08:00
environ.py	Neuron: compiled log-softmax + greedy argmax in the sampler; eager op census (#27)	2026-10-23 09:42:08 +08:00
load_class.py	initial commit	2026-10-20 12:09:42 +08:00
logger.py	initial commit	2026-10-20 12:09:42 +08:00
model_config_loader.py	initial commit	2026-10-20 12:09:42 +08:00
multimodal_tokens.py	initial commit	2026-10-20 12:09:42 +08:00
platform.py	Neuron: static decode graphs, static-shape MoE, trn2 e4m3 fp8 (issues #9, #12, #24)	2026-10-22 11:59:37 +08:00
py.typed	initial commit	2026-10-20 12:09:42 +08:00
sampling_params.py	initial commit	2026-10-20 12:09:42 +08:00
sequence.py	initial commit	2026-10-20 12:09:42 +08:00
shared_memory.py	initial commit	2026-10-20 12:09:42 +08:00
testing.py	initial commit	2026-10-20 12:09:42 +08:00
timer.py	initial commit	2026-10-20 12:09:42 +08:00
utils.py	Neuron basic mode — part 3: end-to-end check vs the official reference	2026-10-20 20:20:02 +08:00