INNER CODE UNIT · Python

measure

FareedKhan-dev/kimi-k3-in-c · tools/devbw.py:41

def measure(path: str, qd: int) -> float:
    fsz = os.path.getsize(path)
    span = max(1, (fsz - SZ) // ALIGN)
    offs = [random.randrange(0, span) * ALIGN for _ in range(N)]
    err = []

    def worker(k: int):
        try:
            fd = os.open(path, os.O_RDONLY | getattr(os, "O_DIRECT", 0))
        except OSError as e:
            err.append(e); return
        try:
            _raw, buf = aligned_buf(SZ)
            mv = memoryview(buf).cast("B")
            for i in range(k, N, qd):
                got, want = 0, SZ
                while got < want:
                    r = os.preadv(fd, [mv[got:want]], offs[i] + got)

View source record →