Skip to content

Commit 57eb129

Browse files
authored
[SYCL][Joint Matrix] Additional devices with Joint Matrix support (#15351)
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.
1 parent 47c0180 commit 57eb129

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

sycl/source/detail/device_info.hpp

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -802,7 +802,9 @@ struct get_device_info_impl<
802802
{16, 16, 32, 0, 0, 0, matrix_type::fp16, matrix_type::fp16,
803803
matrix_type::fp32, matrix_type::fp32},
804804
};
805-
else if (architecture::intel_gpu_pvc == DeviceArch)
805+
else if ((architecture::intel_gpu_pvc == DeviceArch) ||
806+
(architecture::intel_gpu_bmg_g21 == DeviceArch) ||
807+
(architecture::intel_gpu_lnl_m == DeviceArch))
806808
return {
807809
{8, 0, 0, 0, 16, 32, matrix_type::uint8, matrix_type::uint8,
808810
matrix_type::sint32, matrix_type::sint32},
@@ -827,7 +829,8 @@ struct get_device_info_impl<
827829
};
828830
else if ((architecture::intel_gpu_dg2_g10 == DeviceArch) ||
829831
(architecture::intel_gpu_dg2_g11 == DeviceArch) ||
830-
(architecture::intel_gpu_dg2_g12 == DeviceArch))
832+
(architecture::intel_gpu_dg2_g12 == DeviceArch) ||
833+
(architecture::intel_gpu_arl_h == DeviceArch))
831834
return {
832835
{8, 0, 0, 0, 8, 32, matrix_type::uint8, matrix_type::uint8,
833836
matrix_type::sint32, matrix_type::sint32},

0 commit comments

Comments
 (0)