flashinfer.cake_sampling.stage23_variant_flags

flashinfer.cake_sampling.stage23_variant_flags(device: device | int | None = None) → int

Bit mask of the static stage-2/3 features dispatched on device (STAGE23_FEATURE_BITS); 0 = base form (f64 tests, max / min exchange), the form every unlisted architecture runs. Cached per device index: the eager k > 64 path calls this on every launch.