def _concat_kernel_single_input_output_param_size(op: Operator):
    """
    Return the total size (in bytes) of the concat's params.
    We need to adjust this if we change the concatenate op's params.
    """
    inputs = op._attrs["inputs"]
    rank = inputs[0]._rank()
    size_of_one_output_meta = CONCAT_OUTPUT_META_SIZE * rank
    # There are 3 more params, where each takes 8 bytes, so we add 24 more bytes
    total_params_size = CONCAT_INPUT_META_SIZE + size_of_one_output_meta + 24
    _LOGGER.debug(f'concat op {op._attrs["name"]}: {total_params_size=}')
    return total_params_size

    Return the total size (in bytes) of the concat's params.
    We need to adjust this if we change the concatenate op's params.
    
python/aitemplate/compiler/transform/split_large_concat_ops.py