whn09 563c181ecc Neuron: integer mod without torch.remainder in the sampler; sampler cProfile (#27) ... torch.remainder ran ~140 ms a call eagerly on the device and was 554 of the 640 ms a decode step spent sampling. a - (a // b) * b gives the same result for the non-negative operands (checked on 100k random values). Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>		2026-10-23 08:10:16 +08:00
..
tilelang_ops	Neuron: integer mod without torch.remainder in the sampler; sampler cProfile (#27)	2026-10-23 08:10:16 +08:00
__init__.py	initial commit	2026-10-20 12:09:42 +08:00
attention_backend.py	initial commit	2026-10-20 12:09:42 +08:00
layernorm.py	initial commit	2026-10-20 12:09:42 +08:00
linear.py	Neuron: per-layer region graphs + vllm-neuron#71 NKI kernels (issues #10, #11, #13, #25)	2026-10-22 17:36:54 +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	initial commit	2026-10-20 12:09:42 +08:00
vocab_parallel_embedding.py	Neuron: full DeepSeek-V4.1-Flash load + decode region graphs at EP=64 (issues #21, #25)	2026-10-23 01:39:22 +08:00