Merge pull request #985 from 0x12CC/update_root_group_query

Update `root_group` extension tests to use new query API
KhronosGroup · Jan 15, 2025 · c322fdf · c322fdf
2 parents 4b0b671 + ae02230
commit c322fdf
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/extension/oneapi_root_group/root_group.cpp b/tests/extension/oneapi_root_group/root_group.cpp
@@ -31,11 +31,11 @@ static void check_root_group_api() {
   auto bundle =
       sycl::get_kernel_bundle<sycl::bundle_state::executable>(q.get_context());
   auto kernel = bundle.get_kernel<KernelName>();
+  auto local_range = sycl::range<Dimensions>(Dims...);
   auto maxWGs = kernel.template ext_oneapi_get_info<
       sycl::ext::oneapi::experimental::info::kernel_queue_specific::
-          max_num_work_group_sync>(q);
+          max_num_work_groups>(q, local_range, 0);
   REQUIRE(maxWGs >= 1);
-  auto local_range = sycl::range<Dimensions>(Dims...);
   auto global_range = local_range;
   global_range[0] *= maxWGs;
   REQUIRE(global_range.size() == local_range.size() * maxWGs);
@@ -175,11 +175,11 @@ static void check_root_group_barrier() {
   auto bundle =
       sycl::get_kernel_bundle<sycl::bundle_state::executable>(q.get_context());
   auto kernel = bundle.get_kernel<KernelName>();
+  auto local_range = sycl::range<Dimensions>(Dims...);
   auto maxWGs = kernel.template ext_oneapi_get_info<
       sycl::ext::oneapi::experimental::info::kernel_queue_specific::
-          max_num_work_group_sync>(q);
+          max_num_work_groups>(q, local_range, 0);
   REQUIRE(maxWGs >= 1);
-  auto local_range = sycl::range<Dimensions>(Dims...);
   auto global_range = local_range;
   global_range[0] *= maxWGs;
   REQUIRE(global_range.size() == local_range.size() * maxWGs);