def cal_align_ab(x_shape: List[int], dtype="float16") -> int:
    """Returns input alignment."""
    k = x_shape[4]  # CI
    return alignment.find_max_alignment(k, dtype)
Returns input alignment.
python/aitemplate/backend/cuda/conv3d/common_bias.py