Skip to content

Commit

Permalink
[SYCL][Matrix tests] do not exclude SPR from joint_matrix_prefetch te…
Browse files Browse the repository at this point in the history
…st (#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.
  • Loading branch information
dkhaldi committed Mar 5, 2024
1 parent 6e8cdb1 commit f8abcb7
Showing 1 changed file with 4 additions and 3 deletions.
7 changes: 4 additions & 3 deletions sycl/test-e2e/Matrix/joint_matrix_prefetch_impl.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -69,15 +69,16 @@ int main() {
.get_info<sycl::ext::oneapi::experimental::info::device::
matrix_combinations>();
bool support_p = false;
for (int i = 0; i < combinations.size(); i++) {
if (combinations[i].atype == matrix_type::tf32) {
// joint_matrix_prefetch is not supported on DG2
for (unsigned int i = 0; i < combinations.size(); i++) {
if (combinations[i].nsize == 0 || combinations[i].nsize == 16) {
support_p = true;
break;
}
}
if (!support_p) {
std::cout << "Prefetch not supported on this device" << std::endl;
// Once the test is not marke as XFAIL, this should change to return 0;
// Once the test is not marked as XFAIL, this should change to return 0;
return 1;
}
static constexpr size_t M = TM * 2;
Expand Down

0 comments on commit f8abcb7

Please sign in to comment.