| # Copyright 2023 The IREE Authors |
| # |
| # Licensed under the Apache License v2.0 with LLVM Exceptions. |
| # See https://llvm.org/LICENSE.txt for license information. |
| # SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
| |
| from typing import List, TypeVar |
| import numpy as np |
| |
| ArrayLike = TypeVar("ArrayLike") |
| |
| |
| def read_numpy_arrays_from_file(filepath: str) -> List[ArrayLike]: |
| res = [] |
| with open(filepath, "rb") as f: |
| while True: |
| try: |
| res.append(np.load(f)) |
| except EOFError: |
| break |
| return res |
| |
| |
| def write_numpy_arrays_to_file(filepath: str, arrays: List[ArrayLike]): |
| with open(filepath, "wb") as f: |
| for arr in arrays: |
| np.save(f, np.asarray(arr), allow_pickle=False) |