Merge pull request #2566 from RossBrunton/ross/asanfix

ShadowMemory should increment/decrement device handle
oneapi-src · Jan 16, 2025 · 21866c2 · 21866c2
2 parents 45b8e19 + 782a3cf
commit 21866c2
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/source/loader/layers/sanitizer/asan/asan_shadow.hpp b/source/loader/layers/sanitizer/asan/asan_shadow.hpp
@@ -22,9 +22,11 @@ namespace asan {
 
 struct ShadowMemory {
     ShadowMemory(ur_context_handle_t Context, ur_device_handle_t Device)
-        : Context(Context), Device(Device) {}
+        : Context(Context), Device(Device) {
+        urDeviceRetain(Device);
+    }
 
-    virtual ~ShadowMemory() {}
+    virtual ~ShadowMemory() { urDeviceRelease(Device); }
 
     virtual ur_result_t Setup() = 0;