Skip to content

Commit

Permalink
[SYCL][Joint Matrix] Update PVC combinations test and query (#15729)
Browse files Browse the repository at this point in the history 
- update PVC combinations test with the recently added combinations in
https://github.com/intel/llvm/pull/15540/files
- Added two missed combinations for 16x16x16 size to the query
- Fix one typo
  • Loading branch information
@dkhaldi
dkhaldi authored Oct 17, 2024
1 parent e5e0637 commit 4d5c717
Show file tree
Hide file tree
Showing 2 changed files with 91 additions and 1 deletion.
6 changes: 5 additions & 1 deletion sycl/source/detail/device_info.hpp
View file
Original file line number Diff line number Diff line change
Expand Up @@ -856,8 +856,12 @@ struct get_device_info_impl<
matrix_type::fp32, matrix_type::fp16},
{8, 0, 0, 0, 16, 16, matrix_type::fp16, matrix_type::fp16,
matrix_type::fp16, matrix_type::fp16},
{0, 0, 0, 16, 16, 16, matrix_type::fp16, matrix_type::fp16,
matrix_type::fp32, matrix_type::fp32},
{0, 0, 0, 16, 16, 16, matrix_type::fp16, matrix_type::fp16,
matrix_type::fp32, matrix_type::fp16},
{0, 0, 0, 16, 16, 16, matrix_type::fp16, matrix_type::fp16,
matrix_type::fp16, matrix_type::fp32},
{0, 0, 0, 16, 16, 16, matrix_type::fp16, matrix_type::fp16,
matrix_type::fp16, matrix_type::fp16},
{0, 0, 0, 1, 64, 16, matrix_type::fp16, matrix_type::fp16,
Expand All @@ -873,7 +877,7 @@ struct get_device_info_impl<
{0, 0, 0, 32, 64, 16, matrix_type::fp16, matrix_type::fp16,
matrix_type::fp16, matrix_type::fp32},
{0, 0, 0, 32, 64, 16, matrix_type::fp16, matrix_type::fp16,
matrix_type::fp32, matrix_type::bf16},
matrix_type::fp32, matrix_type::fp16},
{0, 0, 0, 32, 64, 16, matrix_type::fp16, matrix_type::fp16,
matrix_type::fp16, matrix_type::fp16},
{0, 0, 0, 1, 64, 32, matrix_type::fp16, matrix_type::fp16,
Expand Down
86 changes: 86 additions & 0 deletions sycl/test-e2e/Matrix/runtime_query_pvc.cpp
View file
Original file line number Diff line number Diff line change
Expand Up @@ -35,14 +35,100 @@ int main() {
matrix_type::sint32, matrix_type::sint32},
{8, 0, 0, 0, 16, 16, matrix_type::fp16, matrix_type::fp16,
matrix_type::fp32, matrix_type::fp32},
{8, 0, 0, 0, 16, 16, matrix_type::fp16, matrix_type::fp16,
matrix_type::fp16, matrix_type::fp32},
{8, 0, 0, 0, 16, 16, matrix_type::fp16, matrix_type::fp16,
matrix_type::fp32, matrix_type::fp16},
{8, 0, 0, 0, 16, 16, matrix_type::fp16, matrix_type::fp16,
matrix_type::fp16, matrix_type::fp16},
{0, 0, 0, 16, 16, 16, matrix_type::fp16, matrix_type::fp16,
matrix_type::fp32, matrix_type::fp32},
{0, 0, 0, 16, 16, 16, matrix_type::fp16, matrix_type::fp16,
matrix_type::fp32, matrix_type::fp16},
{0, 0, 0, 16, 16, 16, matrix_type::fp16, matrix_type::fp16,
matrix_type::fp16, matrix_type::fp32},
{0, 0, 0, 16, 16, 16, matrix_type::fp16, matrix_type::fp16,
matrix_type::fp16, matrix_type::fp16},
{0, 0, 0, 1, 64, 16, matrix_type::fp16, matrix_type::fp16,
matrix_type::fp32, matrix_type::fp32},
{0, 0, 0, 1, 64, 16, matrix_type::fp16, matrix_type::fp16,
matrix_type::fp16, matrix_type::fp32},
{0, 0, 0, 1, 64, 16, matrix_type::fp16, matrix_type::fp16,
matrix_type::fp32, matrix_type::fp16},
{0, 0, 0, 1, 64, 16, matrix_type::fp16, matrix_type::fp16,
matrix_type::fp16, matrix_type::fp16},
{0, 0, 0, 32, 64, 16, matrix_type::fp16, matrix_type::fp16,
matrix_type::fp32, matrix_type::fp32},
{0, 0, 0, 32, 64, 16, matrix_type::fp16, matrix_type::fp16,
matrix_type::fp16, matrix_type::fp32},
{0, 0, 0, 32, 64, 16, matrix_type::fp16, matrix_type::fp16,
matrix_type::fp32, matrix_type::fp16},
{0, 0, 0, 32, 64, 16, matrix_type::fp16, matrix_type::fp16,
matrix_type::fp16, matrix_type::fp16},
{0, 0, 0, 1, 64, 32, matrix_type::fp16, matrix_type::fp16,
matrix_type::fp32, matrix_type::fp32},
{0, 0, 0, 1, 64, 32, matrix_type::fp16, matrix_type::fp16,
matrix_type::fp16, matrix_type::fp32},
{0, 0, 0, 1, 64, 32, matrix_type::fp16, matrix_type::fp16,
matrix_type::fp32, matrix_type::fp16},
{0, 0, 0, 1, 64, 32, matrix_type::fp16, matrix_type::fp16,
matrix_type::fp16, matrix_type::fp16},
{0, 0, 0, 32, 64, 32, matrix_type::fp16, matrix_type::fp16,
matrix_type::fp32, matrix_type::fp32},
{0, 0, 0, 32, 64, 32, matrix_type::fp16, matrix_type::fp16,
matrix_type::fp16, matrix_type::fp32},
{0, 0, 0, 32, 64, 32, matrix_type::fp16, matrix_type::fp16,
matrix_type::fp32, matrix_type::fp16},
{0, 0, 0, 32, 64, 32, matrix_type::fp16, matrix_type::fp16,
matrix_type::fp16, matrix_type::fp16},
{8, 0, 0, 0, 16, 16, matrix_type::bf16, matrix_type::bf16,
matrix_type::bf16, matrix_type::bf16},
{8, 0, 0, 0, 16, 16, matrix_type::bf16, matrix_type::bf16,
matrix_type::fp32, matrix_type::bf16},
{8, 0, 0, 0, 16, 16, matrix_type::bf16, matrix_type::bf16,
matrix_type::bf16, matrix_type::fp32},
{8, 0, 0, 0, 16, 16, matrix_type::bf16, matrix_type::bf16,
matrix_type::fp32, matrix_type::fp32},
{0, 0, 0, 16, 16, 16, matrix_type::bf16, matrix_type::bf16,
matrix_type::fp32, matrix_type::fp32},
{0, 0, 0, 16, 16, 16, matrix_type::bf16, matrix_type::bf16,
matrix_type::bf16, matrix_type::fp32},
{0, 0, 0, 16, 16, 16, matrix_type::bf16, matrix_type::bf16,
matrix_type::fp32, matrix_type::bf16},
{0, 0, 0, 16, 16, 16, matrix_type::bf16, matrix_type::bf16,
matrix_type::bf16, matrix_type::bf16},
{0, 0, 0, 1, 64, 16, matrix_type::bf16, matrix_type::bf16,
matrix_type::fp32, matrix_type::fp32},
{0, 0, 0, 1, 64, 16, matrix_type::bf16, matrix_type::bf16,
matrix_type::bf16, matrix_type::fp32},
{0, 0, 0, 1, 64, 16, matrix_type::bf16, matrix_type::bf16,
matrix_type::fp32, matrix_type::bf16},
{0, 0, 0, 1, 64, 16, matrix_type::bf16, matrix_type::bf16,
matrix_type::bf16, matrix_type::bf16},
{0, 0, 0, 32, 64, 16, matrix_type::bf16, matrix_type::bf16,
matrix_type::fp32, matrix_type::fp32},
{0, 0, 0, 32, 64, 16, matrix_type::bf16, matrix_type::bf16,
matrix_type::bf16, matrix_type::fp32},
{0, 0, 0, 32, 64, 16, matrix_type::bf16, matrix_type::bf16,
matrix_type::fp32, matrix_type::bf16},
{0, 0, 0, 32, 64, 16, matrix_type::bf16, matrix_type::bf16,
matrix_type::bf16, matrix_type::bf16},
{0, 0, 0, 1, 64, 32, matrix_type::bf16, matrix_type::bf16,
matrix_type::fp32, matrix_type::fp32},
{0, 0, 0, 1, 64, 32, matrix_type::bf16, matrix_type::bf16,
matrix_type::bf16, matrix_type::fp32},
{0, 0, 0, 1, 64, 32, matrix_type::bf16, matrix_type::bf16,
matrix_type::fp32, matrix_type::bf16},
{0, 0, 0, 1, 64, 32, matrix_type::bf16, matrix_type::bf16,
matrix_type::bf16, matrix_type::bf16},
{0, 0, 0, 32, 64, 32, matrix_type::bf16, matrix_type::bf16,
matrix_type::fp32, matrix_type::fp32},
{0, 0, 0, 32, 64, 32, matrix_type::bf16, matrix_type::bf16,
matrix_type::bf16, matrix_type::fp32},
{0, 0, 0, 32, 64, 32, matrix_type::bf16, matrix_type::bf16,
matrix_type::fp32, matrix_type::bf16},
{0, 0, 0, 32, 64, 32, matrix_type::bf16, matrix_type::bf16,
matrix_type::bf16, matrix_type::bf16},
{8, 0, 0, 0, 16, 8, matrix_type::tf32, matrix_type::tf32,
matrix_type::fp32, matrix_type::fp32},
};
Expand Down

0 comments on commit 4d5c717

Please sign in to comment.