Add UR_CONTEXT_INFO_REFERENCE_COUNT

Signed-off-by: Jaime Arteaga <[email protected]>

Author: Jaime Arteaga

sycl/plugins/unified_runtime/pi2ur.hpp

@@ -1139,7 +1139,7 @@ inline pi_result piContextGetInfo(pi_context Context, pi_context_info ParamName,
     break;
   }
   case PI_CONTEXT_INFO_REFERENCE_COUNT: {
-    ContextInfoType = UR_EXT_CONTEXT_INFO_REFERENCE_COUNT;
+    ContextInfoType = UR_CONTEXT_INFO_REFERENCE_COUNT;
     break;
   }
   case PI_EXT_ONEAPI_CONTEXT_INFO_USM_FILL2D_SUPPORT: {

sycl/plugins/unified_runtime/ur/adapters/level_zero/ur_level_zero_context.cpp

@@ -101,7 +101,7 @@ UR_APIEXPORT ur_result_t UR_APICALL urContextGetInfo(
     return ReturnValue(&Context->Devices[0], Context->Devices.size());
   case UR_CONTEXT_INFO_NUM_DEVICES:
     return ReturnValue(uint32_t(Context->Devices.size()));
-  case UR_EXT_CONTEXT_INFO_REFERENCE_COUNT:
+  case UR_CONTEXT_INFO_REFERENCE_COUNT:
     return ReturnValue(uint32_t{Context->RefCount.load()});
   case UR_CONTEXT_INFO_USM_MEMCPY2D_SUPPORT:
     // 2D USM memcpy is supported.

sycl/plugins/unified_runtime/ur/ur.hpp

@@ -65,9 +65,6 @@ const int UR_EXT_USM_MEM_FLAG_INITIAL_PLACEMENT_DEVICE = 1 << 28;
 const int UR_EXT_USM_MEM_FLAG_INITIAL_PLACEMENT_HOST = 1 << 29;
 const int UR_EXT_USM_MEM_FLAG_DEVICE_READ_ONLY = 1 << 30;
 
-const ur_context_info_t UR_EXT_CONTEXT_INFO_REFERENCE_COUNT =
-    (ur_context_info_t)(UR_CONTEXT_INFO_FORCE_UINT32 - 2);
-
 const ur_context_info_t UR_EXT_CONTEXT_INFO_ATOMIC_MEMORY_ORDER_CAPABILITIES =
     (ur_context_info_t)(UR_CONTEXT_INFO_FORCE_UINT32 - 1);