Add implementation in clGetMemObjectInfoIntelFPGA and add test for clCreateSubBuffer

Author: sherry-yuan

src/acl_mem.cpp

@@ -416,7 +416,7 @@ CL_API_ENTRY cl_mem clCreateBufferWithPropertiesINTEL(
   cl_bool context_has_device_with_physical_mem;
   unsigned int idevice;
   cl_uint bank_id = 0;
-  unsigned int tmp_mem_id = 0;
+  cl_uint tmp_mem_id = 0;
   acl_lock();
 
 #ifdef MEM_DEBUG_MSG
@@ -433,7 +433,7 @@ CL_API_ENTRY cl_mem clCreateBufferWithPropertiesINTEL(
       bank_id = (cl_uint) * (properties + 1);
     } break;
     case CL_MEM_ALLOC_BUFFER_LOCATION_INTEL: {
-      tmp_mem_id = (unsigned int)*(properties + 1);
+      tmp_mem_id = (cl_uint) * (properties + 1);
     } break;
     default: {
       UNLOCK_BAIL_INFO(CL_INVALID_DEVICE, context, "Invalid properties");
@@ -1376,6 +1376,9 @@ CL_API_ENTRY cl_int CL_API_CALL clGetMemObjectInfoIntelFPGA(
   context = mem->context;
 
   switch (param_name) {
+  case CL_MEM_ALLOC_BUFFER_LOCATION_INTEL:
+    RESULT_UINT(mem->mem_id);
+    break;
   case CL_MEM_TYPE:
     RESULT_ENUM(mem->mem_object_type);
     break;

test/acl_mem_test.cpp

@@ -2815,8 +2815,31 @@ TEST(acl_mem, buffer_location_property) {
   a = clCreateBufferWithPropertiesINTEL(m_context, props, 0, bank_size, 0,
                                         &status);
   CHECK_EQUAL(CL_SUCCESS, status);
+  CHECK_EQUAL(1, acl_ref_count(a));
+  ACL_LOCKED(CHECK(acl_mem_is_valid(a)));
+  cl_uint read_mem_id = 4;
+  size_t size_ret;
+  CHECK_EQUAL(CL_SUCCESS,
+              clGetMemObjectInfo(a, CL_MEM_ALLOC_BUFFER_LOCATION_INTEL,
+                                 sizeof(cl_uint), &read_mem_id, &size_ret));
+  CHECK_EQUAL(0, read_mem_id);
+
+  cl_buffer_region test_region = {0, 2};
+  cl_mem subbuffer =
+      clCreateSubBuffer(a, CL_MEM_READ_WRITE, CL_BUFFER_CREATE_TYPE_REGION,
+                        &test_region, &status);
+  CHECK_EQUAL(CL_SUCCESS, status);
+  CHECK_EQUAL(2, acl_ref_count(a));
+  ACL_LOCKED(CHECK(acl_mem_is_valid(subbuffer)));
+  read_mem_id = 4;
+  CHECK_EQUAL(CL_SUCCESS,
+              clGetMemObjectInfo(subbuffer, CL_MEM_ALLOC_BUFFER_LOCATION_INTEL,
+                                 sizeof(cl_uint), &read_mem_id, &size_ret));
+  CHECK_EQUAL(0, read_mem_id);
+
   ACL_LOCKED(CHECK_EQUAL(acl_bind_buffer_to_device(m_cq->device, a), 1));
 
+  CHECK_EQUAL(CL_SUCCESS, clReleaseMemObject(subbuffer));
   CHECK_EQUAL(CL_SUCCESS, clReleaseMemObject(a));
 }