loopyt 0d144cbce4 feat(ops): report fallbacks at use time, and record why the sampler stays put ... The import-time line only counts kernels the selected backend does not export, which says nothing about whether a running configuration reaches them. `_FallbackNotice` wraps each inherited kernel and prints once, on first call, so an end-to-end run states the answer itself. On 8xH20 with speculative decoding off that answer is a single line: [ops] kernel 'fused_sample_tl' used from 'cuda': 'triton' does not implement it Eighteen of the nineteen kernels on that path are Triton and produce exactly the same tokens as the pure TileLang run. The nineteenth is left there on purpose. A probe of the engine's own sampling parameters (`DEEPINFER_PROBE_SAMPLING=1`, which stages an async D2H copy and reports at exit so it does not synchronise inside the forward) settles what `fused_sample` is being asked to do: temperatures: [1.0, 1.0] top_ps: [0.99, 0.99] top_ks: [-1, -1] That is a real nucleus sample over 4096 candidates, not an argmax, so the draw matters. Two things stand in the way of reproducing it bit for bit, both internal to TileLang: `T.rng_init`/`T.rng_rand` are a Philox stream implemented in its C++ codegen, so a Triton kernel draws a different stream and every sampled token at temperature > 0 would change; and the baseline pins the layout of its LSE accumulator precisely to fix the summation order, after which probabilities are quantised into a 2^31-scaled uint32 CDF where a one-ulp difference can move a column across a boundary. Swapping it anyway would be a behaviour change reported as a kernel swap. The per-kernel fallback exists for exactly this case, so the sampler stays on TileLang and the rest of the path does not. `triton/__init__.py` records this so the gap is not mistaken for an oversight, along with the order in which porting it later would have to proceed: a Philox that matches the stream exactly, then the radix filter, then the fixed-point CDF and the two-level search.		2026-10-22 08:05:05 +08:00
..
asc	initial commit	2026-10-20 12:09:42 +08:00
cuda	initial commit	2026-10-20 12:09:42 +08:00
triton	feat(ops): report fallbacks at use time, and record why the sampler stays put	2026-10-22 08:05:05 +08:00
__init__.py	initial commit	2026-10-20 12:09:42 +08:00
cache_common.py	feat: add SM90 (Hopper) support for DeepSeek-V4.1 inference	2026-10-21 15:18:56 +08:00
candidate_blocks.py	initial commit	2026-10-20 12:09:42 +08:00
common.py	initial commit	2026-10-20 12:09:42 +08:00
compress.py	initial commit	2026-10-20 12:09:42 +08:00
compute_seqlens_k_zos.py	initial commit	2026-10-20 12:09:42 +08:00
conftest.py	initial commit	2026-10-20 12:09:42 +08:00
dflash_copy.py	initial commit	2026-10-20 12:09:42 +08:00
dflash.py	initial commit	2026-10-20 12:09:42 +08:00
flash_mla_utils.py	initial commit	2026-10-20 12:09:42 +08:00
fused_sample.py	feat(ops): report fallbacks at use time, and record why the sampler stays put	2026-10-22 08:05:05 +08:00
fused_transform.py	feat: add SM90 (Hopper) support for DeepSeek-V4.1 inference	2026-10-21 15:18:56 +08:00
gather_engram_tokens_zos.py	initial commit	2026-10-20 12:09:42 +08:00
gather_input_ids_zos.py	initial commit	2026-10-20 12:09:42 +08:00
gather.py	initial commit	2026-10-20 12:09:42 +08:00
get_kv_index.py	initial commit	2026-10-20 12:09:42 +08:00
image_embed_ops.py	initial commit	2026-10-20 12:09:42 +08:00
kv_index_cat.py	initial commit	2026-10-20 12:09:42 +08:00
load_paged_kv_cache.py	initial commit	2026-10-20 12:09:42 +08:00
norm.py	initial commit	2026-10-20 12:09:42 +08:00
ops.py	feat(ops): report fallbacks at use time, and record why the sampler stays put	2026-10-22 08:05:05 +08:00
outlier_detection.py	initial commit	2026-10-20 12:09:42 +08:00
prepare_cp_mirror.py	initial commit	2026-10-20 12:09:42 +08:00
prepare_v4_decode_metadata.py	initial commit	2026-10-20 12:09:42 +08:00
random_sampling.py	initial commit	2026-10-20 12:09:42 +08:00
sampler.py	initial commit	2026-10-20 12:09:42 +08:00
schedule_mtp_varlen.py	initial commit	2026-10-20 12:09:42 +08:00
state_cache.py	initial commit	2026-10-20 12:09:42 +08:00
store_kv_cache.py	initial commit	2026-10-20 12:09:42 +08:00
store_mtp_info.py	initial commit	2026-10-20 12:09:42 +08:00
structure_output.py	initial commit	2026-10-20 12:09:42 +08:00
sum_smaller_probs.py	initial commit	2026-10-20 12:09:42 +08:00
swa_index.py	initial commit	2026-10-20 12:09:42 +08:00