SWDEV-239327 - Remove amd_mem_obj during unregistervar
Change-Id: I2130eaa21369b9634a9459680061138c61eaaaa4
Этот коммит содержится в:
1 родитель
1dc24194a3
Коммит
ee2ff4bc5e
2 files changed
+2
No files matched your search
@@ -245,6 +245,7 @@ public:
|
|||||||
~RegisteredVar() {}
|
~RegisteredVar() {}
|
||||||
|
|
||||||
hipDeviceptr_t getdeviceptr() const { return devicePtr_; };
|
hipDeviceptr_t getdeviceptr() const { return devicePtr_; };
|
||||||
|
amd::Memory* amd_mem_obj() const { return amd_mem_obj_; };
|
||||||
size_t getvarsize() const { return size_; };
|
size_t getvarsize() const { return size_; };
|
||||||
|
|
||||||
size_t size_; // Size of the variable
|
size_t size_; // Size of the variable
|
||||||
|
|||||||
@@ -239,6 +239,7 @@ std::vector< std::pair<hipModule_t, bool> >* PlatformState::unregisterVar(hipMod
|
|||||||
for (size_t dev = 0; dev < g_devices.size(); ++dev) {
|
for (size_t dev = 0; dev < g_devices.size(); ++dev) {
|
||||||
if (dvar.rvars[dev].getdeviceptr()) {
|
if (dvar.rvars[dev].getdeviceptr()) {
|
||||||
amd::MemObjMap::RemoveMemObj(dvar.rvars[dev].getdeviceptr());
|
amd::MemObjMap::RemoveMemObj(dvar.rvars[dev].getdeviceptr());
|
||||||
|
dvar.rvars[dev].amd_mem_obj()->release();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
vars_.erase(it++);
|
vars_.erase(it++);
|
||||||
|
|||||||
Reference in new issue
Block a user