From f8abcb79732bf1f2e30f900a411561f5e028c3bd Mon Sep 17 00:00:00 2001 From: Dounia Khaldi Date: Tue, 5 Mar 2024 10:47:15 -0600 Subject: [PATCH] [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. --- sycl/test-e2e/Matrix/joint_matrix_prefetch_impl.hpp | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/sycl/test-e2e/Matrix/joint_matrix_prefetch_impl.hpp b/sycl/test-e2e/Matrix/joint_matrix_prefetch_impl.hpp index 47355e13f7ae4..f3948a50908bc 100644 --- a/sycl/test-e2e/Matrix/joint_matrix_prefetch_impl.hpp +++ b/sycl/test-e2e/Matrix/joint_matrix_prefetch_impl.hpp @@ -69,15 +69,16 @@ int main() { .get_info(); 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;