[SYCL] Wait on queue in non-uniform-wk-gp-test (#612)

Streams may leak if the associated queue is not synchronized. These
changes forces an explicit wait on queue in non-uniform-wk-gp-test.

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

Author: steffenlarsen

SYCL/Basic/diagnostics/non-uniform-wk-gp-test.cpp

@@ -13,12 +13,13 @@
 using namespace cl::sycl;
 
 int test() {
-  try {
-    queue q = queue();
-    auto device = q.get_device();
-    auto deviceName = device.get_info<cl::sycl::info::device::name>();
-    std::cout << " Device Name: " << deviceName << std::endl;
+  queue q = queue();
+  auto device = q.get_device();
+  auto deviceName = device.get_info<cl::sycl::info::device::name>();
+  std::cout << " Device Name: " << deviceName << std::endl;
 
+  int res = 1;
+  try {
     const int N = 1;
     q.submit([&](handler &cgh) {
       cl::sycl::stream kernelout(108 * 64 + 128, 64, cgh);
@@ -29,23 +30,23 @@ int test() {
                       << cl::sycl::endl;
           });
     });
-
+    std::cout << "Test failed: no exception thrown." << std::endl;
   } catch (sycl::runtime_error &E) {
     if (std::string(E.what()).find(
             "Non-uniform work-groups are not supported by the target device") !=
         std::string::npos) {
       std::cout << E.what() << std::endl;
       std::cout << "Test passed: caught the expected error." << std::endl;
-      return 0;
+      res = 0;
     } else {
       std::cout << E.what() << std::endl;
       std::cout << "Test failed: received error is incorrect." << std::endl;
-      return 1;
     }
   }
+  q.wait();
 
   std::cout << "Test passed: results are correct." << std::endl;
-  return 0;
+  return res;
 }
 
 int main() {