test needs to ensure partition_equally is supported before exercising it

Signed-off-by: Chris Perkins <[email protected]>

Author: cperkinsintel

SYCL/Basic/partition_supported.cpp

@@ -6,12 +6,15 @@
 // Nvidia should not allow sub_devices but does not throw corresponding error.
 // XFAIL: hip_nvidia
 /* Check that:
-1) [info::device::partition_properties]: returns the partition properties
+1) if partition_equally is supported, then we check that the correct 
+invalid errc is returned if more than max_compute_units are requested
+
+2) [info::device::partition_properties]: returns the partition properties
 supported by this SYCL device; a vector of info::partition_property. If this
 SYCL device cannot be partitioned into at least two sub devices then the
 returned vector **must be empty**.
 
-2) [create_sub_devices()]: If the SYCL device
+3) [create_sub_devices()]: If the SYCL device
 does not support info::partition_property::partition_by_affinity_domain or the
 SYCL device does not support the info::partition_affinity_domain provided, an
 exception with the **feature_not_supported error code must be thrown**.
@@ -57,24 +60,10 @@ int main() {
 
   auto dev = cl::sycl::device(cl::sycl::default_selector());
 
-  cl::sycl::info::partition_property partitionProperty =
-      cl::sycl::info::partition_property::partition_by_affinity_domain;
-  cl::sycl::info::partition_affinity_domain affinityDomain =
-      cl::sycl::info::partition_affinity_domain::next_partitionable;
-
-  if (supports_partition_property(dev, partitionProperty)) {
-    if (supports_affinity_domain(dev, partitionProperty, affinityDomain)) {
-      auto subDevices = dev.create_sub_devices<
-          cl::sycl::info::partition_property::partition_by_affinity_domain>(
-          affinityDomain);
-
-      if (subDevices.size() < 2) {
-        std::cerr << "device::create_sub_device(info::partition_affinity_"
-                     "domain) should have returned at least 2 devices"
-                  << std::endl;
-        return -1;
-      }
-    }
+  // check exceed max_compute_units
+  cl::sycl::info::partition_property partitionEqually =
+      cl::sycl::info::partition_property::partition_equally;
+  if (supports_partition_property(dev, partitionEqually)) {
     auto maxUnits = dev.get_info<sycl::info::device::max_compute_units>();
     try {
       std::vector<sycl::device> v = dev.create_sub_devices<
@@ -93,6 +82,27 @@ int main() {
         return -1;
       }
     }
+  }
+
+  cl::sycl::info::partition_property partitionProperty =
+      cl::sycl::info::partition_property::partition_by_affinity_domain;
+  cl::sycl::info::partition_affinity_domain affinityDomain =
+      cl::sycl::info::partition_affinity_domain::next_partitionable;
+
+  if (supports_partition_property(dev, partitionProperty)) {
+    if (supports_affinity_domain(dev, partitionProperty, affinityDomain)) {
+      auto subDevices = dev.create_sub_devices<
+          cl::sycl::info::partition_property::partition_by_affinity_domain>(
+          affinityDomain);
+
+      if (subDevices.size() < 2) {
+        std::cerr << "device::create_sub_device(info::partition_affinity_"
+                     "domain) should have returned at least 2 devices"
+                  << std::endl;
+        return -1;
+      }
+    }
+    
   } else {
     try {
       auto subDevices = dev.create_sub_devices<