github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-04-29 03:00:14 -04:00
Code Issues Packages Projects Releases Wiki Activity

hsa terminate without exceptions (#4006)

Browse Source 
* hsa terminate without exceptions

* cleaner

* linter
This commit is contained in:
nimlgen
2024-03-30 16:03:46 +03:00
committed by GitHub
parent aa76d566c2
commit 478c040e1c
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tinygrad/runtime/ops_hsa.py
View File

@@ -269,8 +269,10 @@ def hsa_terminate():
# Need to stop/delete aql queue before hsa shut down, this leads to gpu hangs.
for dev in HSADevice.devices:
Profiler.process(dev)
setattr(dev, 'synchronize', lambda: None) # some destructors might require to sync, but hw_queue is removed.
del dev.hw_queue
# hsa_shut_down cleans up all hsa-related resources.
hsa.hsa_shut_down()
HSADevice.synchronize = lambda: None #type:ignore
HSAProgram.__del__ = lambda _: None #type:ignore
if Profiler.collected_events: Profiler.save("/tmp/profile.json")