[Matrix] pass "-DtM=16 -DtK=32 -DNCACHE1=32 -DKCACHE1=32" for cpu in …

…joint_matrix_bf16_fill_k_cache.cpp (#12893)
intel · Mar 5, 2024 · d518aa5 · d518aa5
1 parent ed5ff8b
commit d518aa5
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 4 deletions.
diff --git a/sycl/test-e2e/Matrix/SG32/joint_matrix_bf16_fill_k_cache.cpp b/sycl/test-e2e/Matrix/SG32/joint_matrix_bf16_fill_k_cache.cpp
@@ -8,8 +8,11 @@
 // REQUIRES: matrix
 // REQUIRES-INTEL-DRIVER: lin: 27501, win: 101.4943
 
-// RUN: %{build} -o %t.out -ffp-model=precise
-// RUN: %{run} %t.out
+// RUN: %{build} -o %t_gpu.out -ffp-model=precise
+// RUN: %if gpu %{ %{run} %t_gpu.out %}
+
+// RUN: %{build} -ffp-model=precise -o %t_cpu.out -DtM=16 -DtK=32 -DNCACHE1=32 -DKCACHE1=32
+// RUN: %if cpu %{ %{run} %t_cpu.out %}
 
 // -ffp-model=precise is added to not depend on compiler defaults.
 

diff --git a/sycl/test-e2e/Matrix/joint_matrix_bf16_fill_k_cache.cpp b/sycl/test-e2e/Matrix/joint_matrix_bf16_fill_k_cache.cpp
@@ -7,8 +7,11 @@
 //===----------------------------------------------------------------------===//
 // REQUIRES: matrix
 
-// RUN: %{build} -o %t.out -ffp-model=precise
-// RUN: %{run} %t.out
+// RUN: %{build} -o %t_gpu.out -ffp-model=precise
+// RUN: %if gpu %{ %{run} %t_gpu.out %}
+
+// RUN: %{build} -ffp-model=precise -o %t_cpu.out -DtM=16 -DtK=32 -DNCACHE1=32 -DKCACHE1=32
+// RUN: %if cpu %{ %{run} %t_cpu.out %}
 
 // -ffp-model=precise is added to not depend on compiler defaults.