def get_alignments(dtype: str) -> List[int]:
    """
    Return all of the valid alignment values for the dtype.
    """
    dtype = normalize_dtype(dtype)
    if dtype in ("float16", "bfloat16"):
        return [8, 4, 2, 1]
    elif dtype in ("float", "float32"):
        return [4, 2, 1]
    else:
        raise NotImplementedError(f"unsupported {dtype=} for alignments")
Return all of the valid alignment values for the dtype.
python/aitemplate/utils/alignment.py