Skip to content

Commit

Permalink
[SYCL][Joint Matrix] Additional devices with Joint Matrix support (#1…
Browse files Browse the repository at this point in the history
…5351)

New Intel GPU devices with matrix support have been added to
https://github.com/intel/llvm/blob/sycl/sycl/doc/extensions/experimental/sycl_ext_oneapi_device_architecture.asciidoc#new-enumeration-of-architectures,
namely:

- Battlemage and Lunar Lake that share same matrix combinations as PVC
in the matrix query

- Arrow Lake H that shares the same matrix combinations  as DG2

Hence, this PR updates the matrix query to includes these 3 new devices.
  • Loading branch information
dkhaldi authored Sep 12, 2024
1 parent 47c0180 commit 57eb129
Showing 1 changed file with 5 additions and 2 deletions.
7 changes: 5 additions & 2 deletions sycl/source/detail/device_info.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -802,7 +802,9 @@ struct get_device_info_impl<
{16, 16, 32, 0, 0, 0, matrix_type::fp16, matrix_type::fp16,
matrix_type::fp32, matrix_type::fp32},
};
else if (architecture::intel_gpu_pvc == DeviceArch)
else if ((architecture::intel_gpu_pvc == DeviceArch) ||
(architecture::intel_gpu_bmg_g21 == DeviceArch) ||
(architecture::intel_gpu_lnl_m == DeviceArch))
return {
{8, 0, 0, 0, 16, 32, matrix_type::uint8, matrix_type::uint8,
matrix_type::sint32, matrix_type::sint32},
Expand All @@ -827,7 +829,8 @@ struct get_device_info_impl<
};
else if ((architecture::intel_gpu_dg2_g10 == DeviceArch) ||
(architecture::intel_gpu_dg2_g11 == DeviceArch) ||
(architecture::intel_gpu_dg2_g12 == DeviceArch))
(architecture::intel_gpu_dg2_g12 == DeviceArch) ||
(architecture::intel_gpu_arl_h == DeviceArch))
return {
{8, 0, 0, 0, 8, 32, matrix_type::uint8, matrix_type::uint8,
matrix_type::sint32, matrix_type::sint32},
Expand Down

0 comments on commit 57eb129

Please sign in to comment.