mirror of
https://github.com/tinygrad/tinygrad.git
synced 2026-01-09 06:58:11 -05:00
ab3ac2b58d
* use HWInterface in autogen * mockgpu * HWInterface * more HWInterface * fix * fix * old code * fix * implicit field definition * add offset check to mockgpu too * refactor * forgot to pass flags + read rewrite * test * play with vfio * nv: this should be kept * try this * vfio * rm overwrite=True * linetr * do not reinit kfd * minor * mypy * mock * init them once --------- Co-authored-by: patrini32 <patrini23@proton.me>
40 lines
1.3 KiB
Python
40 lines
1.3 KiB
Python
from typing import Any
|
|
from dataclasses import dataclass
|
|
|
|
class VirtFileDesc:
|
|
def __init__(self, fd): self.fd, self.off = fd, 0
|
|
def ioctl(self, fd, req, argp): raise NotImplementedError()
|
|
def mmap(self, st, sz, prot, flags, fd, off): raise NotImplementedError()
|
|
def close(self, fd): return 0
|
|
|
|
class TextFileDesc(VirtFileDesc):
|
|
def __init__(self, fd, text):
|
|
super().__init__(fd)
|
|
self.content = text
|
|
|
|
def ioctl(self, fd, req, argp): return 0
|
|
def read_contents(self, size=None):
|
|
ret = self.content[self.off:self.off+(size or len(self.content))]
|
|
self.off += (size or len(self.content))
|
|
return ret
|
|
def seek(self, offset): self.off += offset
|
|
class DirFileDesc(VirtFileDesc):
|
|
def __init__(self, fd, child_names):
|
|
super().__init__(fd)
|
|
self.child_names = child_names
|
|
|
|
def ioctl(self, fd, req, argp): return 0
|
|
def list_contents(self): return self.child_names
|
|
|
|
@dataclass(frozen=True)
|
|
class VirtFile:
|
|
path: str
|
|
fdcls: Any # TODO: fix this Union[VirtFileDesc, functools.partial[VirtFileDesc]]
|
|
|
|
class VirtDriver:
|
|
def __init__(self):
|
|
self.tracked_files = []
|
|
self.tracked_addresses = []
|
|
def track_address(self, staddr, enaddr, rcb, wcb): self.tracked_addresses.append((staddr, enaddr, rcb, wcb))
|
|
def open(self, name, flags, mode, fdcls): raise NotImplementedError()
|