blob: a2aa0a193785ec998ca776ec15c0d7885d2c220a [file] [log] [blame]
# 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)