from torch._C import _generate_upgraders_bytecode from typing import List def format_bytecode(table): # given a nested tuple, convert it to nested list def listify(content): if not isinstance(content, tuple): return content return [listify(i) for i in content] formatted_table = {} for entry in table: identifier = entry[0] content = entry[1] content = listify(content) formatted_table[identifier] = content return formatted_table def generate_bytecode() -> List: yaml_content = [] upgraders = _generate_upgraders_bytecode() for (upgrader_name, upgrader_bytecode) in upgraders: entry = {upgrader_name: format_bytecode(upgrader_bytecode)} yaml_content.append(entry) return yaml_content if __name__ == "__main__": raise RuntimeError("This file is not meant to be run directly")