trunk/96600b6cc815b5b6c7c8db104fafb6e77dc69e88: [Native DSL] Share topk policy across JIT and AOT (#196599)

发生了什么

Human Notes: Match JIT and AOT conditionals, unify expected coverage. Codex: Make the torch-free AOT declaration the shared source of truth for topk dtype, capability, K, N, and work-rung selection. JIT now imports that policy, gains the measured SM90 and bfloat16 regimes, and uses the same dynamic-N kernel variants as AOT. Replace the exact-N AOT grid with runtime-N radix kernels and one K=16 register branch ladder…

摘要按规则整理自下方来源原文

来源