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

fix hsa sync issue (#3847)

Browse Source 
* fix hsa sync issue

* linter
This commit is contained in:
nimlgen
2024-03-21 04:00:30 +03:00
committed by GitHub
parent f271cd682b
commit 85691c8e20
2 changed files with 23 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tinygrad/runtime/graph/hsa.py
View File

@@ -187,7 +187,7 @@ class HSAGraph(MultiDeviceJITGraph):
# When synchronizing to aql packets, we only need to sync to the latest one, as they are executed in order.
signal_deps, aql_deps = [x for x in rdeps if isinstance(x, hsa.hsa_signal_t)], [x for x in rdeps if isinstance(x, int)]
deps = signal_deps + [max(aql_deps)] if aql_deps else []
deps = signal_deps + ([max(aql_deps)] if len(aql_deps) > 0 else [])
for dep in deps: wait_signals.append(self.dependency_as_signal(dep, sync_with_aql_packets=sync_with_aql_packets))
if new_dependency is not None: