SWDEV-239327 - Remove amd_mem_obj during unregistervar

Change-Id: I2130eaa21369b9634a9459680061138c61eaaaa4


[ROCm/hip commit: 421bc7dfcb]
このコミットが含まれているのは:
kjayapra-amd
2020-06-05 19:28:25 -04:00
committed by Karthik Jayaprakash
コミット 81e6261ff8
2個のファイルの変更2行の追加0行の削除
+1
ファイルの表示
@@ -245,6 +245,7 @@ public:
~RegisteredVar() {}
hipDeviceptr_t getdeviceptr() const { return devicePtr_; };
amd::Memory* amd_mem_obj() const { return amd_mem_obj_; };
size_t getvarsize() const { return size_; };
size_t size_; // Size of the variable
+1
ファイルの表示
@@ -239,6 +239,7 @@ std::vector< std::pair<hipModule_t, bool> >* PlatformState::unregisterVar(hipMod
for (size_t dev = 0; dev < g_devices.size(); ++dev) {
if (dvar.rvars[dev].getdeviceptr()) {
amd::MemObjMap::RemoveMemObj(dvar.rvars[dev].getdeviceptr());
dvar.rvars[dev].amd_mem_obj()->release();
}
}
vars_.erase(it++);