diff --git a/sycl/test-e2e/Basic/subdevice_pi.cpp b/sycl/test-e2e/Basic/subdevice_pi.cpp
index b41c9a21dc055..4a0e4a06e016d 100644
--- a/sycl/test-e2e/Basic/subdevice_pi.cpp
+++ b/sycl/test-e2e/Basic/subdevice_pi.cpp
@@ -39,8 +39,10 @@ static std::vector<device> partition_affinity(device dev) {
 }
 
 static std::vector<device> partition_equally(device dev) {
+  auto maxUnits = dev.get_info<info::device::max_compute_units>();
   std::vector<device> subdevices =
-      dev.create_sub_devices<info::partition_property::partition_equally>(1);
+      dev.create_sub_devices<info::partition_property::partition_equally>(
+          maxUnits / 2);
 
   return subdevices;
 }