[SYCL] Initialize buffer range and size in interop constructor

Signed-off-by: Mariya Podchishchaeva <[email protected]>

Author: Fznamznon

sycl/include/CL/sycl/buffer.hpp

@@ -111,8 +111,13 @@ class buffer {
   template <int N = dimensions, typename = EnableIfOneDimension<N>>
   buffer(cl_mem MemObject, const context &SyclContext,
          event AvailableEvent = {}) {
+
+    size_t BufSize = 0;
+    CHECK_OCL_CODE(clGetMemObjectInfo(MemObject, CL_MEM_SIZE, sizeof(size_t),
+                                      &BufSize, nullptr));
+    Range[0] = BufSize / sizeof(T);
     impl = std::make_shared<detail::buffer_impl<AllocatorT>>(
-        MemObject, SyclContext, AvailableEvent);
+        MemObject, SyclContext, BufSize, AvailableEvent);
   }
 
   buffer(const buffer &rhs) = default;

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

@@ -119,8 +119,9 @@ template <typename AllocatorT> class buffer_impl {
   }
 
   buffer_impl(cl_mem MemObject, const context &SyclContext,
-              event AvailableEvent = {})
-      : OpenCLInterop(true), AvailableEvent(AvailableEvent) {
+              const size_t sizeInBytes, event AvailableEvent = {})
+      : OpenCLInterop(true), SizeInBytes(sizeInBytes),
+        AvailableEvent(AvailableEvent) {
     if (SyclContext.is_host())
       throw cl::sycl::invalid_parameter_error(
           "Creation of interoperability buffer using host context is not "

sycl/test/basic_tests/buffer/buffer_interop.cpp

@@ -16,10 +16,14 @@
 using namespace cl::sycl;
 
 int main() {
+  bool Failed = false;
   {
     const size_t Size = 32;
     int Init[Size] = {5};
     cl_int Error = CL_SUCCESS;
+    cl::sycl::range<1> InteropRange;
+    InteropRange[0] = Size;
+    size_t InteropSize = Size * sizeof(int);
 
     queue MyQueue;
 
@@ -29,6 +33,19 @@ int main() {
     CHECK_OCL_CODE(Error);
     buffer<int, 1> Buffer(OpenCLBuffer, MyQueue.get_context());
 
+    if (Buffer.get_range() != InteropRange) {
+          assert(false);
+          Failed = true;
+    }
+    if (Buffer.get_size() != InteropSize) {
+          assert(false);
+          Failed = true;
+    }
+    if (Buffer.get_count() != Size) {
+          assert(false);
+          Failed = true;
+    }
+
     MyQueue.submit([&](handler &CGH) {
       auto B = Buffer.get_access<access::mode::write>(CGH);
       CGH.parallel_for<class BufferInterop>(
@@ -58,8 +75,9 @@ int main() {
         std::cout << " array[" << i << "] is " << Result[i] << " expected "
                   << 20 << std::endl;
         assert(false);
+        Failed = true;
       }
     }
   }
-  return 0;
+  return Failed;
 }