hcq cache invalidation for beam (#5630)

* nv full cache invalidation

* the same command on amd

* linter

* fix amd

* nv no hardcoded consts

* beam default
This commit is contained in:
nimlgen
2024-07-22 18:13:17 +03:00
committed by GitHub
parent c64e9591e3
commit 08a9c0ae5e
4 changed files with 15 additions and 2 deletions

View File

@@ -492,3 +492,8 @@ class AMDDevice(HCQCompiled):
self.kernargs_ptr = self.kernargs.va_addr
if self.timeline_value > (1 << 31): self._wrap_timeline_signal()
if PROFILE: self._prof_process_events()
def invalidate_cache(self):
AMDComputeQueue().memory_barrier().signal(self.timeline_signal, self.timeline_value).submit(self)
self.timeline_value += 1
self.synchronize()