Function update_code

generate_code.py:192–211

replace the text surrounded by "label start" and "label end" to new_code :param file: the .cpp or .h file :param label: the label surrounds the text to be replaced

(file: str, label: str, reformat: bool=True)

def update_code(file: str, label: str, reformat: bool=True) -> None:
    """
    replace the text surrounded by "label start" and "label end" to new_code
    :param file: the .cpp or .h file
    :param label: the label surrounds the text to be replaced
    """
    global str_io
    with open(file, 'r') as f:
        s = f.read()
        start = '// {} start\n'.format(label)
        idx1 = s.find(start) + len(start)
        end = '// {} end'.format(label)
        idx2 = s.find(end)
        assert start in s and end in s
    with open(file, 'w') as f:
        new_s = s[:idx1] + str_io.getvalue() + s[idx2:]
        f.write(new_s)
    str_io = io.StringIO()
    if reformat:
        clang_format(file)


def generate_onnx_converter():