github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-09 23:18:04 -05:00
Code Issues Packages Projects Releases Wiki Activity

hive_reset for mi350 (#13746)

Browse Source
This commit is contained in:
nimlgen
2025-12-18 12:02:28 +03:00
committed by GitHub
parent ceff388f3d
commit 77191fb744
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
extra/amdpci/hive_reset.py
View File

@@ -5,7 +5,7 @@ from tinygrad.runtime.support.system import System, PCIDevice, PCIDevImplBase
from tinygrad.runtime.support.am.amdev import AMDev from tinygrad.runtime.support.am.amdev import AMDev
if __name__ == "__main__": if __name__ == "__main__":
gpus = System.pci_scan_bus(0x1002, [(0xffff, [0x74a1])]) 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(f"reset:{gpu}", gpu, bars=[0, 2, 5]) for gpu in gpus]
amdevs = [] amdevs = []
with Context(DEBUG=2): with Context(DEBUG=2):