loopyt 5e2838aa5e feat(triton): reproduce cuRAND's Philox stream, so the sampler can be ported ... The last blocker for `fused_sample` was the random stream: `T.rng_init`/`T.rng_rand` lower to `curand_init`/`curand` with `curandStatePhilox4_32_10_t`, and the seeding is not documented. Its mixing function is public but `curand_init` folds the seed, the subsequence and the offset into the counter in a way that has to be measured rather than derived. Measured (`scripts/h20_rng_dump.py` dumps the real stream), the layout is: key = (seed & 0xFFFFFFFF, seed >> 32) ctr = (offset & 0xFFFFFFFF, offset >> 32, subsequence & 0xFFFFFFFF, subsequence >> 32) with draws taking the block's four words in order before `ctr.x` advances. The subsequence sits in the third and fourth counter words, not the first -- a cuRAND subsequence spans 2^64 blocks. That detail is invisible at subsequence 0 and wrong everywhere else, which is why the check covers both. `scripts/h20_philox_check.py` compares our kernel against the dumped reference over five seeds (including 2^64-1) and four subsequences (0, 1, 2048, 4097 -- the last two are the actual `sample_positions` from a run): 20/20 streams bit-identical. One trap worth noting: the multipliers and the second Weyl constant all exceed int32, so referencing them as Python ints inside the kernel silently wraps them negative. They are materialised as uint32 tiles instead. This does not yet change any output -- `fused_sample` is still the TileLang kernel. It removes the reason it had to be, which is the point.		2026-10-22 08:37:08 +08:00
..
ab_cases.py	feat(triton): port fused_transform, compress, prepare_v4 and both metadata kernels	2026-10-22 07:47:24 +08:00
build_vendor.sh	initial commit	2026-10-20 12:09:42 +08:00
build.sh	initial commit	2026-10-20 12:09:42 +08:00
compare_layers.py	fix(sm90): expand expert scales on the contraction axis only	2026-10-21 16:50:31 +08:00
format_csrc.sh	initial commit	2026-10-20 12:09:42 +08:00
gen_stub.sh	initial commit	2026-10-20 12:09:42 +08:00
h20_ask.py	style(sm90): satisfy ruff and add the missing type annotations	2026-10-21 17:09:08 +08:00
h20_attn_check.py	style(sm90): satisfy ruff and add the missing type annotations	2026-10-21 17:09:08 +08:00
h20_engram_check.py	style(sm90): satisfy ruff and add the missing type annotations	2026-10-21 17:09:08 +08:00
h20_expert_weight_check.py	style(sm90): satisfy ruff and add the missing type annotations	2026-10-21 17:09:08 +08:00
h20_index_decode_check.py	test(sm90): rule out the KV cache and the decode indexer	2026-10-21 20:02:19 +08:00
h20_kv_roundtrip.py	test(sm90): rule out the KV cache and the decode indexer	2026-10-21 20:02:19 +08:00
h20_ops_ab.py	feat(triton): complete P1 with norm, lookup and metadata kernels	2026-10-22 00:28:38 +08:00
h20_philox_check.py	feat(triton): reproduce cuRAND's Philox stream, so the sampler can be ported	2026-10-22 08:37:08 +08:00
h20_quant_check.py	test(sm90): decode FP4 by table lookup in the quant check	2026-10-21 17:15:16 +08:00
h20_real_requant_check.py	style(sm90): satisfy ruff and add the missing type annotations	2026-10-21 17:09:08 +08:00
h20_recipe_probe.py	debug: capture final logits, and keep the profile forward out of the dumps	2026-10-21 17:44:21 +08:00
h20_rng_dump.py	feat(triton): reproduce cuRAND's Philox stream, so the sampler can be ported	2026-10-22 08:37:08 +08:00
h20_smoke_check.py	style(sm90): satisfy ruff and add the missing type annotations	2026-10-21 17:09:08 +08:00
h20_topk_check.py	style(sm90): satisfy ruff and add the missing type annotations	2026-10-21 17:09:08 +08:00
ref_layer_dump.py	debug: capture final logits, and keep the profile forward out of the dumps	2026-10-21 17:44:21 +08:00
ref_next_token.py	debug: per-step decode dumps, and prove the flip is decode-only	2026-10-21 19:07:49 +08:00