[SYCL] add device_type to SYCL_PI_TRACE and device_selector exception

Author: mkrainiuk

sycl/source/device_selector.cpp

@@ -106,6 +106,25 @@ device select_device(DSelectorInvocableType DeviceSelectorInvocable,
     return *res;
   }
 
+  auto Selector = DeviceSelectorInvocable.target<int (*)(const sycl::device &)>();
+  if ((Selector && *Selector == gpu_selector_v)
+          || DeviceSelectorInvocable.target<sycl::gpu_selector>()) {
+    throw sycl::runtime_error(
+        "No device of requested type 'info::device_type::gpu' available.",
+        PI_ERROR_DEVICE_NOT_FOUND);
+  }
+  if ((Selector && *Selector == cpu_selector_v)
+          || DeviceSelectorInvocable.target<sycl::cpu_selector>()) {
+    throw sycl::runtime_error(
+        "No device of requested type 'info::device_type::cpu' available.",
+        PI_ERROR_DEVICE_NOT_FOUND);
+  }
+  if ((Selector && *Selector == accelerator_selector_v)
+          || DeviceSelectorInvocable.target<sycl::accelerator_selector>()) {
+    throw sycl::runtime_error("No device of requested type "
+                              "'info::device_type::accelerator' available.",
+                              PI_ERROR_DEVICE_NOT_FOUND);
+  }
   throw sycl::runtime_error("No device of requested type available.",
                             PI_ERROR_DEVICE_NOT_FOUND);
 }
@@ -137,10 +156,20 @@ select_device(const DSelectorInvocableType &DeviceSelectorInvocable,
 /// 2. CPU
 /// 3. Host
 /// 4. Accelerator
+
+static void traceDeviceSelector(const std::string &DeviceType) {
+  bool ShouldTrace = false;
+  ShouldTrace = detail::pi::trace(detail::pi::TraceLevel::PI_TRACE_BASIC);
+  if (ShouldTrace) {
+    std::cout << "SYCL_PI_TRACE[all]: Requested device_type: " << DeviceType << std::endl;
+  }
+}
+
 __SYCL_EXPORT int default_selector_v(const device &dev) {
   // The default selector doesn't reject any devices.
   int Score = 0;
 
+  traceDeviceSelector("info::device_type::automatic");
   if (dev.get_info<info::device::device_type>() == detail::get_forced_type())
     Score += 2000;
 
@@ -165,6 +194,7 @@ __SYCL_EXPORT int default_selector_v(const device &dev) {
 __SYCL_EXPORT int gpu_selector_v(const device &dev) {
   int Score = detail::REJECT_DEVICE_SCORE;
 
+  traceDeviceSelector("info::device_type::gpu");
   if (dev.is_gpu()) {
     Score = 1000;
     Score += detail::getDevicePreference(dev);
@@ -175,6 +205,7 @@ __SYCL_EXPORT int gpu_selector_v(const device &dev) {
 __SYCL_EXPORT int cpu_selector_v(const device &dev) {
   int Score = detail::REJECT_DEVICE_SCORE;
 
+  traceDeviceSelector("info::device_type::cpu");
   if (dev.is_cpu()) {
     Score = 1000;
     Score += detail::getDevicePreference(dev);
@@ -185,6 +216,7 @@ __SYCL_EXPORT int cpu_selector_v(const device &dev) {
 __SYCL_EXPORT int accelerator_selector_v(const device &dev) {
   int Score = detail::REJECT_DEVICE_SCORE;
 
+  traceDeviceSelector("info::device_type::accelerator");
   if (dev.is_accelerator()) {
     Score = 1000;
     Score += detail::getDevicePreference(dev);
@@ -196,6 +228,7 @@ int host_selector::operator()(const device &dev) const {
   // Host device has been removed and host_selector has been deprecated, so this
   // should never be able to select a device.
   std::ignore = dev;
+  traceDeviceSelector("info::device_type::host");
   return detail::REJECT_DEVICE_SCORE;
 }