Added public C functions to access data payload of USMMemory* objects (…

…#541)
IntelPython · Aug 24, 2021 · 260573c · 260573c
1 parent 58f78f5
commit 260573c
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/dpctl/memory/_memory.pyx b/dpctl/memory/_memory.pyx
@@ -706,3 +706,13 @@ def as_usm_memory(obj):
             "USM allocation represented by argument {}".
             format(obj)
         )
+
+
+cdef public DPCTLSyclUSMRef get_usm_pointer(_Memory obj):
+    return obj.memory_ptr
+
+cdef public DPCTLSyclContextRef get_context(_Memory obj):
+    return obj.queue._context.get_context_ref()
+
+cdef public size_t get_nbytes(_Memory obj):
+    return <size_t>obj.nbytes