Skip to content

Commit e198d84

Browse files
committed
move ManagedQueue
1 parent 726435d commit e198d84

File tree

3 files changed

+26
-23
lines changed

3 files changed

+26
-23
lines changed

source/loader/layers/sanitizer/asan_interceptor.cpp

Lines changed: 0 additions & 23 deletions
Original file line numberDiff line numberDiff line change
@@ -51,29 +51,6 @@ constexpr auto kSPIR_DeviceSanitizerReportMem = "__DeviceSanitizerReportMem";
5151
constexpr auto kSPIR_AsanDeviceGlobalCount = "__AsanDeviceGlobalCount";
5252
constexpr auto kSPIR_AsanDeviceGlobalMetadata = "__AsanDeviceGlobalMetadata";
5353

54-
struct ManagedQueue {
55-
ManagedQueue(ur_context_handle_t Context, ur_device_handle_t Device) {
56-
[[maybe_unused]] auto Result = context.urDdiTable.Queue.pfnCreate(
57-
Context, Device, nullptr, &Handle);
58-
assert(Result == UR_RESULT_SUCCESS);
59-
}
60-
61-
~ManagedQueue() {
62-
[[maybe_unused]] auto Result =
63-
context.urDdiTable.Queue.pfnRelease(Handle);
64-
assert(Result == UR_RESULT_SUCCESS);
65-
}
66-
67-
// Disable copy semantics
68-
ManagedQueue(const ManagedQueue &) = delete;
69-
ManagedQueue &operator=(const ManagedQueue &) = delete;
70-
71-
operator ur_queue_handle_t() { return Handle; }
72-
73-
private:
74-
ur_queue_handle_t Handle = nullptr;
75-
};
76-
7754
uptr MemToShadow_CPU(uptr USM_SHADOW_BASE, uptr UPtr) {
7855
return USM_SHADOW_BASE + (UPtr >> 3);
7956
}

source/loader/layers/sanitizer/ur_sanitizer_utils.cpp

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,18 @@
1515

1616
namespace ur_sanitizer_layer {
1717

18+
ManagedQueue::ManagedQueue(ur_context_handle_t Context,
19+
ur_device_handle_t Device) {
20+
[[maybe_unused]] auto Result =
21+
context.urDdiTable.Queue.pfnCreate(Context, Device, nullptr, &Handle);
22+
assert(Result == UR_RESULT_SUCCESS);
23+
}
24+
25+
ManagedQueue::~ManagedQueue() {
26+
[[maybe_unused]] auto Result = context.urDdiTable.Queue.pfnRelease(Handle);
27+
assert(Result == UR_RESULT_SUCCESS);
28+
}
29+
1830
ur_context_handle_t GetContext(ur_queue_handle_t Queue) {
1931
ur_context_handle_t Context{};
2032
[[maybe_unused]] auto Result = context.urDdiTable.Queue.pfnGetInfo(

source/loader/layers/sanitizer/ur_sanitizer_utils.hpp

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,20 @@
1616

1717
namespace ur_sanitizer_layer {
1818

19+
struct ManagedQueue {
20+
ManagedQueue(ur_context_handle_t Context, ur_device_handle_t Device);
21+
~ManagedQueue();
22+
23+
// Disable copy semantics
24+
ManagedQueue(const ManagedQueue &) = delete;
25+
ManagedQueue &operator=(const ManagedQueue &) = delete;
26+
27+
operator ur_queue_handle_t() { return Handle; }
28+
29+
private:
30+
ur_queue_handle_t Handle = nullptr;
31+
};
32+
1933
ur_context_handle_t GetContext(ur_queue_handle_t Queue);
2034
ur_context_handle_t GetContext(ur_program_handle_t Program);
2135
ur_device_handle_t GetDevice(ur_queue_handle_t Queue);

0 commit comments

Comments
 (0)