Merge pull request #843 from frasercrmck/fix-reqd-work-group-size

Fix reqd_work_group_size test
KhronosGroup · Jan 11, 2024 · 53238ed · 53238ed
2 parents 9691bc0 + 84089ee
commit 53238ed
Showing 1 changed file with 13 additions and 3 deletions.
diff --git a/tests/optional_kernel_features/kernel_features_reqd_work_group_size.cpp b/tests/optional_kernel_features/kernel_features_reqd_work_group_size.cpp
@@ -104,9 +104,19 @@ void test_size() {
         is_exception_expected, expected_errc, queue);
   }
   {
-    RUN_SUBMISSION_CALL_ND_RANGE(
-        N, Dimensions, is_exception_expected, expected_errc, queue,
-        [[sycl::reqd_work_group_size(N)]], kname, NO_KERNEL_BODY);
+    if constexpr (Dimensions == 1) {
+      RUN_SUBMISSION_CALL_ND_RANGE(
+          N, Dimensions, is_exception_expected, expected_errc, queue,
+          [[sycl::reqd_work_group_size(N)]], kname, NO_KERNEL_BODY);
+    } else if constexpr (Dimensions == 2) {
+      RUN_SUBMISSION_CALL_ND_RANGE(
+          N, Dimensions, is_exception_expected, expected_errc, queue,
+          [[sycl::reqd_work_group_size(N, N)]], kname, NO_KERNEL_BODY);
+    } else {
+      RUN_SUBMISSION_CALL_ND_RANGE(
+          N, Dimensions, is_exception_expected, expected_errc, queue,
+          [[sycl::reqd_work_group_size(N, N, N)]], kname, NO_KERNEL_BODY);
+    }
   }
 }
 #endif  // !SYCL_CTS_COMPILING_WITH_HIPSYCL