Skip to content

Commit f8abcb7

Browse files
authored
[SYCL][Matrix tests] do not exclude SPR from joint_matrix_prefetch test (#12876)
The test how it was written excluded SPR (tf32 type is not supported there). But joint_matrix_prefetch should be supported on SPR so here I change the way to enable this test by only excluding DG2.
1 parent 6e8cdb1 commit f8abcb7

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

sycl/test-e2e/Matrix/joint_matrix_prefetch_impl.hpp

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -69,15 +69,16 @@ int main() {
6969
.get_info<sycl::ext::oneapi::experimental::info::device::
7070
matrix_combinations>();
7171
bool support_p = false;
72-
for (int i = 0; i < combinations.size(); i++) {
73-
if (combinations[i].atype == matrix_type::tf32) {
72+
// joint_matrix_prefetch is not supported on DG2
73+
for (unsigned int i = 0; i < combinations.size(); i++) {
74+
if (combinations[i].nsize == 0 || combinations[i].nsize == 16) {
7475
support_p = true;
7576
break;
7677
}
7778
}
7879
if (!support_p) {
7980
std::cout << "Prefetch not supported on this device" << std::endl;
80-
// Once the test is not marke as XFAIL, this should change to return 0;
81+
// Once the test is not marked as XFAIL, this should change to return 0;
8182
return 1;
8283
}
8384
static constexpr size_t M = TM * 2;

0 commit comments

Comments
 (0)