Skip to content

Commit cdd75df

Browse files
authored
Merge pull request intel#1539 from myler/cherry-pick-4d1861bf
[SYCL]Add check on subgroup size test to make sure its supported on platform (intel#1466)
2 parents beb3c57 + d2a0a0a commit cdd75df

File tree

1 file changed

+19
-3
lines changed

1 file changed

+19
-3
lines changed

SYCL/SubGroupMask/GroupSize.cpp

Lines changed: 19 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -82,9 +82,25 @@ int main() {
8282
#ifdef SYCL_EXT_ONEAPI_SUB_GROUP_MASK
8383
queue Queue;
8484

85-
test<8>(Queue);
86-
test<16>(Queue);
87-
test<32>(Queue);
85+
for (const auto &x :
86+
Queue.get_device().get_info<info::device::sub_group_sizes>()) {
87+
88+
switch (x) {
89+
case 8:
90+
test<8>(Queue);
91+
break;
92+
case 16:
93+
test<16>(Queue);
94+
break;
95+
case 32:
96+
test<32>(Queue);
97+
break;
98+
default:
99+
std::cout << "Sub group size of " << x << " supported, but not tested."
100+
<< std::endl;
101+
break;
102+
}
103+
}
88104

89105
std::cout << "Test passed." << std::endl;
90106
#else

0 commit comments

Comments
 (0)