def _split_kernel_single_input_output_param_size(op: Operator):
    """
    Return the total size (in bytes) of the split's params.
    We need to adjust this if we change the split op's params.
    Note this is conservative by multiplying input_meta and constant 24 bytes.
    """
    outputs = op._attrs["outputs"]
    rank = outputs[0]._rank()
    size_of_input_meta = SPLIT_INPUT_META_SIZE * rank
    # There are 3 more params, where each takes 8 bytes, so we add 24 more bytes
    total_params_size = SPLIT_OUTPUT_META_SIZE + size_of_input_meta + 24
    _LOGGER.debug(f'split op op._attrs["name"]: {total_params_size=}')
    return total_params_size

    Return the total size (in bytes) of the split's params.
    We need to adjust this if we change the split op's params.
    Note this is conservative by multiplying input_meta and constant 24 bytes.
    
python/aitemplate/compiler/transform/split_large_split_ops.py