SWDEV-276396 - Implement hipDeviceReset

Add a Purge function to MemObjMap Change-Id: Iac51dfda9a7b7c45f2f4a0dc35f7a623121aba1a [ROCm/clr commit: d13581efa7]
2021-06-07 14:14:08 -07:00
parent a9abd850ea
commit b8de199652
2 changed files with 15 additions and 0 deletions
@@ -327,6 +327,20 @@ void MemObjMap::UpdateAccess(amd::Device *peerDev) {
  }
 }

+void MemObjMap::Purge(amd::Device* dev) {
+  assert(dev != nullptr);
+
+  amd::ScopedLock lock(AllocatedLock_);
+  for (auto it = MemObjMap_.cbegin() ; it != MemObjMap_.cend() ;) {
+    const std::vector<Device*>& devices = it->second->getContext().devices();
+    if (devices.size() == 1 && devices[0] == dev) {
+      it = MemObjMap_.erase(it);
+    } else {
+      ++it;
+    }
+  }
+}
+
 Device::BlitProgram::~BlitProgram() {
  if (program_ != nullptr) {
    program_->release();