| 76 | |
| 77 | @dataclass(frozen=True, eq=True) |
| 78 | class BufferSpec: |
| 79 | # TODO: move device, size, dtype here? |
| 80 | uncached: bool = False |
| 81 | cpu_access: bool = False |
| 82 | host: bool = False |
| 83 | nolru: bool = False |
| 84 | external_ptr: int|None = None |
| 85 | |
| 86 | class MultiBuffer: |
| 87 | def __init__(self, device:tuple[str, ...], size:int, dtype:DType): |
no outgoing calls
searching dependent graphs…