[SYCL] Fix set_final_data(..) for std::vector<bool> iterator (#3099)

Fixed set_final_data() taking std::vector<bool> which was not compiling because
such a vector has no data method. The patch adds a special case for bool type.

Test: intel/llvm-test-suite#122

Signed-off-by: mdimakov [email protected]

Author: maximdimakov

sycl/include/CL/sycl/detail/sycl_mem_obj_t.hpp

@@ -20,6 +20,7 @@
 #include <CL/sycl/stl.hpp>
 
 #include <cstring>
+#include <memory>
 #include <type_traits>
 
 __SYCL_INLINE_NAMESPACE(cl) {
@@ -51,7 +52,13 @@ class __SYCL_EXPORT SYCLMemObjT : public SYCLMemObjI {
 
   template <typename T>
   using EnableIfOutputIteratorT = enable_if_t<
-      /*is_output_iterator<T>::value &&*/ !std::is_pointer<T>::value>;
+      /*is_output_iterator<T>::value &&*/ !std::is_pointer<T>::value &&
+      !std::is_same<typename T::value_type, bool>::value>;
+
+  template <typename T>
+  using EnableIfOutputIteratorBool =
+      enable_if_t<!std::is_pointer<T>::value &&
+                  std::is_same<typename T::value_type, bool>::value>;
 
   template <typename T>
   using EnableIfDefaultAllocator =
@@ -184,6 +191,22 @@ class __SYCL_EXPORT SYCLMemObjT : public SYCLMemObjI {
     };
   }
 
+  template <typename Destination>
+  __SYCL_DLL_LOCAL EnableIfOutputIteratorBool<Destination>
+  set_final_data(Destination FinalData) {
+    MUploadDataFunctor = [this, FinalData]() {
+      using DestinationValueT = iterator_value_type_t<Destination>;
+      // TODO if Destination is ContiguousIterator then don't create
+      // ContiguousStorage. updateHostMemory works only with pointer to
+      // continuous data.
+      const size_t Size = MSizeInBytes / sizeof(DestinationValueT);
+      std::unique_ptr<bool[]> ContiguousStorage(new bool[Size]);
+      updateHostMemory(ContiguousStorage.get());
+      std::copy(ContiguousStorage.get(), ContiguousStorage.get() + Size,
+                FinalData);
+    };
+  }
+
 protected:
   void updateHostMemory(void *const Ptr);