def normalize_dtype(dtype: str) -> str:
    """Returns a normalized dtype str.

    Parameters
    ----------
    dtype: str
        A data type string.

    Returns
    ----------
    str
        normalized dtype str.
    """
    if dtype == "int":
        return "int32"
    if dtype == "float":
        return "float32"
    return dtype
Returns a normalized dtype str.

Parameters
----------
dtype: str
A data type string.

Returns
----------
str
normalized dtype str.
python/aitemplate/compiler/dtype.py