def _cal_align_ab(x_shape: List[int], dtype="float16") -> int:
    """Returns input alignment."""
    k = x_shape[3]  # CI
    return alignment.find_max_alignment(k, dtype)
Returns input alignment.
python/aitemplate/backend/cuda/conv2d/common.py