[SYCL][XPTI] Add instrumentation for USM allocations (#5391)

Save code location information about USM allocations to TLS.

Author: Alexander Batashev

sycl/include/CL/sycl/detail/usm_impl.hpp

@@ -7,6 +7,7 @@
 // ===--------------------------------------------------------------------=== //
 #pragma once
 
+#include <CL/sycl/detail/common.hpp>
 #include <CL/sycl/detail/export.hpp>
 #include <CL/sycl/usm/usm_enums.hpp>
 
@@ -17,13 +18,16 @@ namespace usm {
 
 __SYCL_EXPORT void *alignedAlloc(size_t Alignment, size_t Bytes,
                                  const context &Ctxt, const device &Dev,
-                                 cl::sycl::usm::alloc Kind);
+                                 cl::sycl::usm::alloc Kind,
+                                 const code_location &CL);
 
 __SYCL_EXPORT void *alignedAllocHost(size_t Alignment, size_t Bytes,
                                      const context &Ctxt,
-                                     cl::sycl::usm::alloc Kind);
+                                     cl::sycl::usm::alloc Kind,
+                                     const code_location &CL);
 
-__SYCL_EXPORT void free(void *Ptr, const context &Ctxt);
+__SYCL_EXPORT void free(void *Ptr, const context &Ctxt,
+                        const code_location &CL);
 
 } // namespace usm
 } // namespace detail

sycl/include/CL/sycl/usm.hpp

@@ -8,6 +8,7 @@
 #pragma once
 
 #include <CL/sycl/context.hpp>
+#include <CL/sycl/detail/common.hpp>
 #include <CL/sycl/detail/export.hpp>
 #include <CL/sycl/device.hpp>
 #include <CL/sycl/exception.hpp>
@@ -24,8 +25,10 @@ namespace sycl {
 __SYCL_EXPORT void *aligned_alloc(size_t alignment, size_t size,
                                   const device &dev, const context &ctxt,
                                   usm::alloc kind,
-                                  const property_list &propList);
-__SYCL_EXPORT void free(void *ptr, const context &ctxt);
+                                  const property_list &propList,
+                                  const detail::code_location CL);
+__SYCL_EXPORT void free(void *ptr, const context &ctxt,
+                        const detail::code_location CL);
 
 template <typename T, usm::alloc AllocKind, size_t Alignment = alignof(T)>
 class usm_allocator {
@@ -74,11 +77,12 @@ class usm_allocator {
   /// Allocates memory.
   ///
   /// \param NumberOfElements is a count of elements to allocate memory for.
-  T *allocate(size_t NumberOfElements) {
+  T *allocate(size_t NumberOfElements, const detail::code_location CL =
+                                           detail::code_location::current()) {
 
     auto Result = reinterpret_cast<T *>(
         aligned_alloc(getAlignment(), NumberOfElements * sizeof(value_type),
-                      MDevice, MContext, AllocKind, MPropList));
+                      MDevice, MContext, AllocKind, MPropList, CL));
     if (!Result) {
       throw memory_allocation_error();
     }
@@ -89,9 +93,11 @@ class usm_allocator {
   ///
   /// \param Ptr is a pointer to memory being deallocated.
   /// \param Size is a number of elements previously passed to allocate.
-  void deallocate(T *Ptr, size_t) {
+  void deallocate(
+      T *Ptr, size_t,
+      const detail::code_location CL = detail::code_location::current()) {
     if (Ptr) {
-      free(Ptr, MContext);
+      free(Ptr, MContext, CL);
     }
   }