[UR][L0] Make the queue [root-]device appear first in the context (#11213)

Signed-off-by: smaslov <[email protected]>

Author: smaslov-intel

sycl/plugins/unified_runtime/ur/adapters/level_zero/queue.cpp

@@ -279,6 +279,35 @@ UR_APIEXPORT ur_result_t UR_APICALL urQueueCreate(
     ur_queue_handle_t
         *Queue ///< [out] pointer to handle of queue object created
 ) {
+
+  // Make the Device appear as the first device in the context since this
+  // is where the urProgramBuild will only build the module to. Also, if
+  // the Device is a sub-device then see if there is a also its root-device
+  // in the context and make that go first instead (because sub-device can
+  // run code built for its root-device).
+  //
+  // TODO: this is all hacky and should be removed when we add support
+  // for building to all the devices in the context.
+  //
+  { // Lock context for thread-safe update
+    std::scoped_lock Lock(Context->Mutex);
+    UR_ASSERT(Context->isValidDevice(Device), UR_RESULT_ERROR_INVALID_DEVICE);
+
+    auto MakeFirst = Context->Devices.begin();
+    for (auto I = Context->Devices.begin(); I != Context->Devices.end(); ++I) {
+      if (*I == Device) {
+        MakeFirst = I;
+        if (!Device->RootDevice)
+          break;
+        // continue the search for possible root-device in the context
+      } else if (*I == Device->RootDevice) {
+        MakeFirst = I;
+        break; // stop the search
+      }
+    }
+    if (MakeFirst != Context->Devices.begin())
+      std::iter_swap(MakeFirst, Context->Devices.begin());
+  }
   ur_queue_flags_t Flags{};
   if (Props) {
     Flags = Props->flags;
@@ -297,8 +326,6 @@ UR_APIEXPORT ur_result_t UR_APICALL urQueueCreate(
     }
   }
 
-  UR_ASSERT(Context->isValidDevice(Device), UR_RESULT_ERROR_INVALID_DEVICE);
-
   // Create placeholder queues in the compute queue group.
   // Actual L0 queues will be created at first use.
   std::vector<ze_command_queue_handle_t> ZeComputeCommandQueues(