def gen_int_var_min_max(
    values: List[int], name: str = None, symbolic_value: Optional[sympy.Basic] = None
):
    """
    A helper function to generate IntImm or IntVar depending on the length of values.
    Only keeps [min, max] pairs if there are more than 2 values.
    """
    return gen_int_var(
        [min(values), max(values)], name=name, symbolic_value=symbolic_value
    )
A helper function to generate IntImm or IntVar depending on the length of values.
Only keeps [min, max] pairs if there are more than 2 values.
python/aitemplate/utils/shape_utils.py