test: update system memory test(s)

Signed-off-by: Narendra Bagria <[email protected]>

Author: narenbagria

tests/gtests/ocl/api/test_memory_usm.cpp

@@ -24,6 +24,7 @@
 
 #include <cstdint>
 #include <vector>
+#include <CL/cl_ext.h>
 
 using namespace dnnl::impl::xpu::ocl;
 
@@ -204,8 +205,26 @@ TEST(ocl_memory_usm_test, ErrorMakeMemoryUsingSystemMemory) {
     memory::desc mem_d({n}, memory::data_type::f32, memory::format_tag::x);
 
     std::vector<float> system_buf(n);
-
-    if (eng.get()->mayiuse_system_memory_allocators()) {
+    bool system_memory_supported = false;
+
+#ifdef cl_intel_unified_shared_memory
+    cl_device_id device = ocl_interop::get_device(eng);
+    cl_device_unified_shared_memory_capabilities_intel
+            system_memory_capabilities_intel
+            = 0;
+
+    ASSERT_EQ(
+            clGetDeviceInfo(device,
+                    CL_DEVICE_SHARED_SYSTEM_MEM_CAPABILITIES_INTEL,
+                    sizeof(cl_device_unified_shared_memory_capabilities_intel),
+                    &system_memory_capabilities_intel, nullptr),
+            CL_SUCCESS);
+
+    system_memory_supported = system_memory_capabilities_intel
+            & CL_UNIFIED_SHARED_MEMORY_ACCESS_INTEL;
+#endif
+
+    if (system_memory_supported) {
         EXPECT_NO_THROW(
                 memory mem = ocl_interop::make_memory(mem_d, eng,
                         ocl_interop::memory_kind::usm, system_buf.data()));