def find_tile_size(k: int) -> int:
    """
    Find the smalled m that would make m * k multiples of 8.
    For odd k, only apply for k <= 5
    """
    m = 1 if k > 1 else 8
    for i in (1, 2, 4, 8):
        if (k * i) % 8 == 0:
            m = i
            break
    if k % 2 == 1 and k >= 7:
        m = 1
    return m

    Find the smalled m that would make m * k multiples of 8.
    For odd k, only apply for k <= 5
    
python/aitemplate/backend/cuda/softmax/softmax.py