Function _log2

src/fused_mm_sampling/tl_argsort.py:94–100

(i)

@constexpr_function
def _log2(i):
    log2 = 0
    n = i
    while n > 1:
        n >>= 1
        log2 += 1
    return log2


_get_int_dtype = constexpr_function(core.get_int_dtype)