[UR] Proposal of new API for memory object properties

Signed-off-by: Tikhomirova, Kseniya <kseniya.tikhomirova@intel.com>

Author: KseniyaTikhomirova

include/ur.py

@@ -2242,9 +2242,9 @@ class ur_program_dditable_t(Structure):
 ###############################################################################
 ## @brief Function-pointer for urKernelSetArgMemObj
 if __use_win_types:
-    _urKernelSetArgMemObj_t = WINFUNCTYPE( ur_result_t, ur_kernel_handle_t, c_ulong, ur_mem_handle_t )
+    _urKernelSetArgMemObj_t = WINFUNCTYPE( ur_result_t, ur_kernel_handle_t, c_ulong, ur_mem_handle_t, const ur_mem_obj_properties_t* )
 else:
-    _urKernelSetArgMemObj_t = CFUNCTYPE( ur_result_t, ur_kernel_handle_t, c_ulong, ur_mem_handle_t )
+    _urKernelSetArgMemObj_t = CFUNCTYPE( ur_result_t, ur_kernel_handle_t, c_ulong, ur_mem_handle_t, const ur_mem_obj_properties_t* )
 
 ###############################################################################
 ## @brief Function-pointer for urKernelSetSpecializationConstants

include/ur_api.h

@@ -252,6 +252,7 @@ typedef enum ur_structure_type_t {
     UR_STRUCTURE_TYPE_SAMPLER_NATIVE_PROPERTIES = 24,       ///< ::ur_sampler_native_properties_t
     UR_STRUCTURE_TYPE_QUEUE_NATIVE_DESC = 25,               ///< ::ur_queue_native_desc_t
     UR_STRUCTURE_TYPE_DEVICE_PARTITION_PROPERTIES = 26,     ///< ::ur_device_partition_properties_t
+    UR_STRUCTURE_TYPE_MEM_OBJ_PROPERTIES = 27,              ///< ::ur_mem_obj_properties_t
     /// @cond
     UR_STRUCTURE_TYPE_FORCE_UINT32 = 0x7fffffff
     /// @endcond
@@ -3837,6 +3838,15 @@ urKernelSetArgSampler(
     ur_sampler_handle_t hArgValue ///< [in] handle of Sampler object.
 );
 
+///////////////////////////////////////////////////////////////////////////////
+/// @brief Memory object properties
+typedef struct ur_mem_obj_properties_t {
+    ur_structure_type_t stype; ///< [in] type of this structure, must be
+                               ///< ::UR_STRUCTURE_TYPE_MEM_OBJ_PROPERTIES
+    void *pNext;               ///< [in,out][optional] pointer to extension-specific structure
+    ur_mem_flags_t memory_access;    ///< [in] Memory access flag. Allowed values are: UR_MEM_FLAG_READ_WRITE, UR_MEM_FLAG_WRITE_ONLY, UR_MEM_FLAG_READ_ONLY.
+} ur_mem_obj_properties_t;
+
 ///////////////////////////////////////////////////////////////////////////////
 /// @brief Set a Memory object as the argument value of a Kernel.
 ///
@@ -3856,7 +3866,8 @@ UR_APIEXPORT ur_result_t UR_APICALL
 urKernelSetArgMemObj(
     ur_kernel_handle_t hKernel, ///< [in] handle of the kernel object
     uint32_t argIndex,          ///< [in] argument index in range [0, num args - 1]
-    ur_mem_handle_t hArgValue   ///< [in][optional] handle of Memory object.
+    ur_mem_handle_t hArgValue,   ///< [in][optional] handle of Memory object.
+    const ur_mem_obj_properties_t* pProperties ///< [in][optional] pointer to memory object usage properties.
 );
 
 ///////////////////////////////////////////////////////////////////////////////
@@ -6461,6 +6472,7 @@ typedef struct ur_kernel_set_arg_mem_obj_params_t {
     ur_kernel_handle_t *phKernel;
     uint32_t *pargIndex;
     ur_mem_handle_t *phArgValue;
+    const ur_mem_obj_properties_t* pProperties;
 } ur_kernel_set_arg_mem_obj_params_t;
 
 ///////////////////////////////////////////////////////////////////////////////

include/ur_ddi.h

@@ -509,7 +509,8 @@ typedef ur_result_t(UR_APICALL *ur_pfnKernelSetArgSampler_t)(
 typedef ur_result_t(UR_APICALL *ur_pfnKernelSetArgMemObj_t)(
     ur_kernel_handle_t,
     uint32_t,
-    ur_mem_handle_t);
+    ur_mem_handle_t,
+    const ur_mem_obj_properties_t*);
 
 ///////////////////////////////////////////////////////////////////////////////
 /// @brief Function-pointer for urKernelSetSpecializationConstants