[SYCL] Enable set_final_data for shared_ptr

Signed-off-by: Vladimir Lazarev <[email protected]>

Author: vladimirlaz

sycl/include/CL/sycl/detail/buffer_impl.hpp

@@ -227,6 +227,15 @@ template <typename AllocatorT> class buffer_impl : public SYCLMemObjT {
     };
   }
 
+  template <template <typename T> class C, typename T>
+  void set_final_data(
+      C<T> FinalData,
+      typename std::enable_if<
+          std::is_convertible<C<T>, weak_ptr_class<T>>::value>::type * = 0) {
+    weak_ptr_class<T> WeakFinalData(FinalData);
+    set_final_data(WeakFinalData);
+  }
+
   template <typename Destination>
   void set_final_data(
       Destination FinalData,

sycl/test/basic_tests/buffer/buffer.cpp

@@ -191,6 +191,22 @@ int main() {
       std::weak_ptr<int> resultWeak = result;
       Buffer.set_final_data(resultWeak);
       queue myQueue;
+      myQueue.submit([&](handler &cgh) {
+        auto B = Buffer.get_access<access::mode::write>(cgh);
+        cgh.parallel_for<class weakPointer>(
+            range<1>{10}, [=](id<1> index) { B[index] = 1; });
+      });
+    }
+    checkAllOf(result.get(), 10, 1, __LINE__);
+  }
+
+  {
+    int data[10] = {0};
+    std::shared_ptr<int> result(new int[10]());
+    {
+      buffer<int, 1> Buffer(data, range<1>(10));
+      Buffer.set_final_data(result);
+      queue myQueue;
       myQueue.submit([&](handler &cgh) {
         auto B = Buffer.get_access<access::mode::write>(cgh);
         cgh.parallel_for<class sharedPointer>(