github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-02-15 17:15:48 -05:00
Code Issues Packages Projects Releases Wiki Activity

start hcq docs (#5411)

Browse Source 
* start hcq docs

* more hcq docs

* docs

* docs

* linter

* correct args

* linter

* ts returns int
This commit is contained in:
nimlgen
2024-07-15 21:31:11 +03:00
committed by GitHub
parent 9a7d5a148e
commit c9ec7ce070
7 changed files with 371 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
tinygrad/runtime/ops_amd.py
View File

@@ -410,13 +410,13 @@ class AMDDevice(HCQCompatCompiled):
kio.free_memory_of_gpu(self.kfd, handle=mem.handle)
@classmethod
def _read_signal(self, sig): return sig.value
def _read_signal(self, signal): return signal.value
@classmethod
def _read_timestamp(self, sig): return sig.start_ts
def _read_timestamp(self, signal): return signal.start_ts
@classmethod
def _set_signal(self, sig, value): sig.value = value
def _set_signal(self, signal, value): signal.value = value
@classmethod
def _alloc_signal(self, value=0, **kwargs) -> hsa.amd_signal_t:
@@ -428,7 +428,7 @@ class AMDDevice(HCQCompatCompiled):
return ret
@classmethod
def _free_signal(self, sig): self.signals_pool.append(sig)
def _free_signal(self, signal): self.signals_pool.append(signal)
@classmethod
def _wait_signal(self, signal:hsa.amd_signal_t, value=0, timeout=10000):