tinygrad/tinygrad
2
0
Fork 3
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-08-29 09:56:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

hive_reset respects lock (#14618)

Browse Source
This commit is contained in:
nimlgen
2026-02-08 10:47:25 +03:00
committed by GitHub
parent b10802eb53
commit e29a88ca09
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
extra/amdpci/hive_reset.py
+1 -1
View File
@@ -6,7 +6,7 @@ from tinygrad.runtime.support.am.amdev import AMDev
if __name__ == "__main__":
gpus = System.pci_scan_bus(0x1002, [(0xffff, [0x74a1, 0x75a0])])
pcidevs = [PCIDevice(f"reset:{gpu}", gpu, bars=[0, 2, 5]) for gpu in gpus]
pcidevs = [PCIDevice("AM", gpu, bars=[0, 2, 5]) for gpu in gpus]
amdevs = []
with Context(DEBUG=2):
for pcidev in pcidevs: