[SYCL] Add support for USM aspects. (#2706)

This patch adds support for the USM-related aspects and the custom aspect.
All device aspects defined by the SYCL 2020 Provisional Specification are
now supported.

Signed-off-by: Gail Lyons <[email protected]>

Author: glyons-intel

sycl/source/detail/device_impl.cpp

@@ -225,6 +225,8 @@ bool device_impl::has(aspect Aspect) const {
     return is_gpu();
   case aspect::accelerator:
     return is_accelerator();
+  case aspect::custom:
+    return false;
   case aspect::fp16:
     return has_extension("cl_khr_fp16");
   case aspect::fp64:
@@ -241,6 +243,16 @@ bool device_impl::has(aspect Aspect) const {
     return get_info<info::device::is_linker_available>();
   case aspect::queue_profiling:
     return get_info<info::device::queue_profiling>();
+  case aspect::usm_device_allocations:
+    return get_info<info::device::usm_device_allocations>();
+  case aspect::usm_host_allocations:
+    return get_info<info::device::usm_host_allocations>();
+  case aspect::usm_shared_allocations:
+    return get_info<info::device::usm_shared_allocations>();
+  case aspect::usm_restricted_shared_allocations:
+    return get_info<info::device::usm_restricted_shared_allocations>();
+  case aspect::usm_system_allocator:
+    return get_info<info::device::usm_system_allocator>();
   default:
     throw runtime_error("This device aspect has not been implemented yet.",
                         PI_INVALID_DEVICE);

sycl/test/basic_tests/aspects.cpp

@@ -25,16 +25,19 @@ int main() {
     pltIdx++;
     if (plt.has(aspect::host)) {
       std::cout << "Platform #" << pltIdx
-                << " type: host supports:" << std::endl;
+                << " type: Host supports:" << std::endl;
     } else if (plt.has(aspect::cpu)) {
       std::cout << "Platform #" << pltIdx
-                << " type: cpu supports:" << std::endl;
+                << " type: CPU supports:" << std::endl;
     } else if (plt.has(aspect::gpu)) {
       std::cout << "Platform #" << pltIdx
-                << " type: gpu supports:" << std::endl;
+                << " type: GPU supports:" << std::endl;
     } else if (plt.has(aspect::accelerator)) {
       std::cout << "Platform #" << pltIdx
-                << " type: accelerator supports:" << std::endl;
+                << " type: Accelerator supports:" << std::endl;
+    } else if (plt.has(aspect::custom)) {
+      std::cout << "Platform #" << pltIdx
+                << " type: Custom supports:" << std::endl;
     } else {
       failed = true;
       std::cout << "Failed: platform #" << pltIdx << " type: unknown"
@@ -66,6 +69,21 @@ int main() {
     if (plt.has(aspect::queue_profiling)) {
       std::cout << "  queue profiling" << std::endl;
     }
+    if (plt.has(aspect::usm_device_allocations)) {
+      std::cout << "  USM allocations" << std::endl;
+    }
+    if (plt.has(aspect::usm_host_allocations)) {
+      std::cout << "  USM host allocations" << std::endl;
+    }
+    if (plt.has(aspect::usm_shared_allocations)) {
+      std::cout << "  USM shared allocations" << std::endl;
+    }
+    if (plt.has(aspect::usm_restricted_shared_allocations)) {
+      std::cout << "  USM restricted shared allocations" << std::endl;
+    }
+    if (plt.has(aspect::usm_system_allocator)) {
+      std::cout << "  USM system allocator" << std::endl;
+    }
   }
   std::cout << "Passed." << std::endl;
   return 0;