Skip to content

Commit 45e7ba6

Browse files
committed
restructure combinations so it can be reused and augmented by similar architectures
1 parent 2c15e2e commit 45e7ba6

File tree

1 file changed

+7
-6
lines changed

1 file changed

+7
-6
lines changed

sycl/source/detail/device_info.hpp

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -838,8 +838,8 @@ struct get_device_info_impl<
838838
};
839839
else if ((architecture::intel_gpu_pvc == DeviceArch) ||
840840
(architecture::intel_gpu_bmg_g21 == DeviceArch) ||
841-
(architecture::intel_gpu_lnl_m == DeviceArch))
842-
return {
841+
(architecture::intel_gpu_lnl_m == DeviceArch)) {
842+
std::vector<ext::oneapi::experimental::matrix::combination> pvc_combs = {
843843
{8, 0, 0, 0, 16, 32, matrix_type::uint8, matrix_type::uint8,
844844
matrix_type::sint32, matrix_type::sint32},
845845
{8, 0, 0, 0, 16, 32, matrix_type::uint8, matrix_type::sint8,
@@ -943,10 +943,11 @@ struct get_device_info_impl<
943943
{8, 0, 0, 0, 16, 8, matrix_type::tf32, matrix_type::tf32,
944944
matrix_type::fp32, matrix_type::fp32},
945945
};
946-
else if ((architecture::intel_gpu_dg2_g10 == DeviceArch) ||
947-
(architecture::intel_gpu_dg2_g11 == DeviceArch) ||
948-
(architecture::intel_gpu_dg2_g12 == DeviceArch) ||
949-
(architecture::intel_gpu_arl_h == DeviceArch))
946+
return pvc_combs;
947+
} else if ((architecture::intel_gpu_dg2_g10 == DeviceArch) ||
948+
(architecture::intel_gpu_dg2_g11 == DeviceArch) ||
949+
(architecture::intel_gpu_dg2_g12 == DeviceArch) ||
950+
(architecture::intel_gpu_arl_h == DeviceArch))
950951
return {
951952
{8, 0, 0, 0, 8, 32, matrix_type::uint8, matrix_type::uint8,
952953
matrix_type::sint32, matrix_type::sint32},

0 commit comments

Comments
 (0)