Skip to content

Commit

Permalink
[SYCL Joint Matrix][E2E] Add 1d vector load/store tests for big combi…
Browse files Browse the repository at this point in the history
…nations (intel#13814)

This change is partially addressing requirement to enable 1d block load
testing (IGC_JointMatrixLoadStoreOpt=2) for only big combinations
32x64x16, 1x64x16, 16x16x16.
This environment variable was added to as minimum tests as possible,
while keeping the coverage of big combinations to maximum.
This environment variable was not added to Matrix/element_wise_ops.cpp
yet, because, IGC compiler needs to be fixed to pass this test with
IGC_JointMatrixLoadStoreOpt=2
  • Loading branch information
YuriPlyakhin authored May 22, 2024
1 parent d39563a commit 306e5ff
Show file tree
Hide file tree
Showing 3 changed files with 4 additions and 0 deletions.
1 change: 1 addition & 0 deletions sycl/test-e2e/Matrix/joint_matrix_bfloat16_packedB.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -10,6 +10,7 @@

// RUN: %{build} -o %t.out
// RUN: %{run} %t.out
// RUN: %if gpu %{ env IGC_JointMatrixLoadStoreOpt=2 %{run} %t.out %}

#include "common.hpp"
#include "joint_matrix_bfloat16_packedB_impl.hpp"
1 change: 1 addition & 0 deletions sycl/test-e2e/Matrix/joint_matrix_rowmajorA_rowmajorB.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,7 @@

// RUN: %{build} -o %t.out
// RUN: %{run} %t.out
// RUN: %if gpu %{ env IGC_JointMatrixLoadStoreOpt=2 %{run} %t.out %}

// This tests support of row major layout for matrix B which does automatic VNNI
// transform. This is currently only available on AMX and XMX of PVC
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -123,6 +123,8 @@ int main() {
int32_t>();

if (combination.nsize == 16) { // architecture::intel_gpu_pvc
res += gemm_row_major<16, 16, 16, class bf16_16x16x16, bfloat16,
bfloat16, float>();
res += gemm_row_major<1, 64, 16, class bf16_1x64x16, bfloat16, bfloat16,
float>();
res += gemm_row_major<32, 64, 16, class bf16_32x64x16, bfloat16,
Expand Down

0 comments on commit 306e5ff

Please sign in to comment.