MCPcopy Index your code
hub / github.com/tinygrad/tinygrad / list_devices

Method list_devices

tinygrad/runtime/support/system.py:79–81  ·  view source on GitHub ↗
(self, vendor:int, devices:tuple[tuple[int, tuple[int, ...]], ...], base_class:int|None=None)

Source from the content-addressed store, hash-verified

77
78 @functools.cache
79 def list_devices(self, vendor:int, devices:tuple[tuple[int, tuple[int, ...]], ...], base_class:int|None=None):
80 if getenv("REMOTE", ""): return [(functools.partial(RemotePCIDevice,sock=s), x) for s,x in RemotePCIDevice.remote_list(vendor,devices,base_class)]
81 return [(APLRemotePCIDevice if OSX else PCIDevice, x) for x in System.pci_scan_bus(vendor, devices, base_class)]
82
83 def pci_probe_device(self, device:str, dev_id:int, vendor:int, devices:tuple[tuple[int, tuple[int, ...]], ...], base_class:int|None=None):
84 try: cl, pcibus = (ds:=hcq_filter_visible_devices(self.list_devices(vendor, devices, base_class), device))[dev_id]

Callers 2

pci_probe_deviceMethod · 0.95
__init__Method · 0.45

Calls 3

getenvFunction · 0.90
remote_listMethod · 0.80
pci_scan_busMethod · 0.80

Tested by

no test coverage detected