mirrors/tinygrad
2
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-24 02:14:17 +00:00
Code Issues Projects Releases Packages Wiki Activity

am: minor parts (#8507)

Browse source
This commit is contained in:
nimlgen 2025-01-05 23:05:21 +03:00 • committed by GitHub
commit b4f4a3ac12
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 1 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

1
tinygrad/runtime/ops_amd.py
View file

@ -183,6 +183,7 @@ class AMDCopyQueue(HWQueue):
if not AMDDevice.driverless and (dev:=signal.timeline_for_device) is not None:
self.q(amd_gpu.SDMA_OP_FENCE | amd_gpu.SDMA_PKT_FENCE_HEADER_MTYPE(3), *data64_le(dev.queue_event_mailbox_ptr), dev.queue_event.event_id)
self.q(amd_gpu.SDMA_OP_TRAP, amd_gpu.SDMA_PKT_TRAP_INT_CONTEXT_INT_CONTEXT(dev.queue_event.event_id))
elif AMDDevice.driverless: self.q(amd_gpu.SDMA_OP_TRAP, amd_gpu.SDMA_PKT_TRAP_INT_CONTEXT_INT_CONTEXT(0))
return self

2
tinygrad/runtime/support/am/amdev.py
View file

@ -203,9 +203,7 @@ class AMMemoryManager:
if AM_DEBUG >= 3: print(f"\tnptes={n_ptes:#x} incr={pte_covers:#x} upd_flags={pt.get_entry(pte_st_idx):#x} frags={frags_cnt:#x}")
# Invalidate TLB after mappings.
self.adev.gmc.flush_tlb(ip="GC", vmid=0, flush_type=2)
self.adev.gmc.flush_tlb(ip="GC", vmid=0)
self.adev.gmc.flush_tlb(ip="MM", vmid=0, flush_type=2)
self.adev.gmc.flush_tlb(ip="MM", vmid=0)
def unmap_range(self, vaddr:int, size:int, free_paddrs=True):