modify context

Signed-off-by: rehana begam <[email protected]>

Author: rbegam

SYCL/Basic/subsubdevice.cpp

@@ -90,7 +90,8 @@ int main() {
                     info::partition_affinity_domain::numa);
         std::cout
             << "Created " << SubSubDevicesDomainNuma.size()
-            << " sub-subdevices using partition by numa affinity domain scheme."
+            << " sub-subdevices from subdevice 0 using partition by numa "
+               "affinity domain scheme."
             << std::endl;
       } catch (feature_not_supported) {
         // okay skip it

SYCL/Basic/subsubdevice_pi.cpp

@@ -2,9 +2,6 @@
 // RUN: %GPU_RUN_PLACEHOLDER SYCL_PI_TRACE=2 %t.out separate affinity %GPU_CHECK_PLACEHOLDER --check-prefix CHECK-SEPARATE
 // RUN: %GPU_RUN_PLACEHOLDER SYCL_PI_TRACE=2 %t.out shared affinity %GPU_CHECK_PLACEHOLDER --check-prefix CHECK-SHARED --implicit-check-not piContextCreate --implicit-check-not piMemBufferCreate
 // RUN: %GPU_RUN_PLACEHOLDER SYCL_PI_TRACE=2 %t.out fused affinity %GPU_CHECK_PLACEHOLDER --check-prefix CHECK-FUSED --implicit-check-not piContextCreate --implicit-check-not piMemBufferCreate
-//
-// Intel OpenCL CPU Runtime supports device partition on all (multi-core)
-// platforms. Other devices may not support this.
 
 #include <CL/sycl.hpp>
 #include <string>
@@ -52,7 +49,7 @@ static bool check_separate(device dev, buffer<int, 1> buf,
 
   log_pi("Create sub sub devices of sub device 0");
   std::vector<device> subsubdevices = partition(subdevices[0]);
-  assert(subsubdevices.size() >= 1);
+  assert(subsubdevices.size() > 1);
   // CHECK-SEPARATE: Create sub sub devices of sub device 0
   // CHECK-SEPARATE: ---> piDevicePartition
 
@@ -81,7 +78,7 @@ static bool check_shared_context(device dev, buffer<int, 1> buf,
 
   log_pi("Create sub sub devices of sub device 0");
   std::vector<device> subsubdevices = partition(subdevices[0]);
-  assert(subsubdevices.size() >= 1);
+  assert(subsubdevices.size() > 1);
   // CHECK-SHARED: Create sub sub devices of sub device 0
   // CHECK-SHARED: ---> piDevicePartition
 
@@ -96,8 +93,8 @@ static bool check_shared_context(device dev, buffer<int, 1> buf,
 
   log_pi("Test sub sub device 0");
   {
-    queue q0(shared_context, subsubdevices[0]);
-    use_mem(buf, q0);
+    queue q(shared_context, subsubdevices[0]);
+    use_mem(buf, q);
   }
   // CHECK-SHARED: Test sub sub device 0
   // CHECK-SHARED: ---> piQueueCreate
@@ -107,7 +104,18 @@ static bool check_shared_context(device dev, buffer<int, 1> buf,
   // see --implicit-check-not above.
   //
   // CHECK-SHARED: ---> piEnqueueKernelLaunch
-  // CHECK-SHARED: ---> piEventsWait
+  // CHECK-SHARED: ---> piQueueFinish
+
+  log_pi("Test sub sub device 1");
+  {
+    queue q1(shared_context, subsubdevices[1]);
+    use_mem(buf, q1);
+  }
+  // CHECK-SHARED: Test sub sub device 1
+  // CHECK-SHARED: ---> piQueueCreate
+  // CHECK-SHARED: ---> piEnqueueKernelLaunch
+  // CHECK-SHARED: ---> piQueueFinish
+  // CHECK-SHARED: ---> piEnqueueMemBufferRead
 
   return true;
 }
@@ -122,7 +130,7 @@ static bool check_fused_context(device dev, buffer<int, 1> buf,
 
   log_pi("Create sub sub devices of sub device 0");
   std::vector<device> subsubdevices = partition(subdevices[0]);
-  assert(subsubdevices.size() >= 1);
+  assert(subsubdevices.size() > 1);
   // CHECK-FUSED: Create sub sub devices of sub device 0
   // CHECK-FUSED: ---> piDevicePartition