[SYCL] Change online compiler test to build IL based on queue (#1028)

Level-zero expects a single device when creating a module. This means
that the online compiler test for level-zero has an assertion on the
supplied context assuring only a single device. However, on multi-GPU
systems this assertion will fail, in turn causing the test to report a
failure. Rather than skipping the test on multi-GPU systems, these
changes makes the test use the device in the queue rather than the first
device in the associated context.

Signed-off-by: Larsen, Steffen <[email protected]>

Author: steffenlarsen

SYCL/OnlineCompiler/online_compiler_L0.cpp

@@ -23,7 +23,7 @@
 using byte = unsigned char;
 
 #ifdef RUN_KERNELS
-sycl::kernel getSYCLKernelWithIL(sycl::context &Context,
+sycl::kernel getSYCLKernelWithIL(sycl::queue &Queue,
                                  const std::vector<byte> &IL) {
 
   ze_module_desc_t ZeModuleDesc = {};
@@ -33,8 +33,8 @@ sycl::kernel getSYCLKernelWithIL(sycl::context &Context,
   ZeModuleDesc.pBuildFlags = "";
   ZeModuleDesc.pConstants = nullptr;
 
-  assert(Context.get_devices().size() == 1 && "Expected to have only 1 device");
-  sycl::device Device = Context.get_devices()[0];
+  sycl::context Context = Queue.get_context();
+  sycl::device Device = Queue.get_device();
   auto ZeDevice = Device.get_native<sycl::backend::ext_oneapi_level_zero>();
   auto ZeContext = Context.get_native<sycl::backend::ext_oneapi_level_zero>();
 

SYCL/OnlineCompiler/online_compiler_OpenCL.cpp

@@ -19,8 +19,10 @@
 using byte = unsigned char;
 
 #ifdef RUN_KERNELS
-sycl::kernel getSYCLKernelWithIL(sycl::context &Context,
+sycl::kernel getSYCLKernelWithIL(sycl::queue &Queue,
                                  const std::vector<byte> &IL) {
+  sycl::context Context = Queue.get_context();
+
   cl_int Err;
   cl_program ClProgram =
       clCreateProgramWithIL(sycl::get_native<sycl::backend::opencl>(Context),

SYCL/OnlineCompiler/online_compiler_common.hpp

@@ -51,7 +51,6 @@ void testSyclKernel(sycl::queue &Q, sycl::kernel Kernel) {
 
 int main(int argc, char **argv) {
   cl::sycl::queue Q;
-  cl::sycl::context Context = Q.get_context();
   cl::sycl::device Device = Q.get_device();
 
   { // Compile and run a trivial OpenCL kernel.
@@ -73,7 +72,7 @@ int main(int argc, char **argv) {
       return 1;
     }
 #ifdef RUN_KERNELS
-    testSyclKernel(Q, getSYCLKernelWithIL(Context, IL));
+    testSyclKernel(Q, getSYCLKernelWithIL(Q, IL));
 #endif // RUN_KERNELS
   }
 
@@ -93,7 +92,7 @@ int main(int argc, char **argv) {
       return 1;
     }
 #ifdef RUN_KERNELS
-    testSyclKernel(Q, getSYCLKernelWithIL(Context, IL));
+    testSyclKernel(Q, getSYCLKernelWithIL(Q, IL));
 #endif // RUN_KERNELS
   }