Namespaces
	precision

Classes
struct	combination

struct	joint_matrix

struct	matrix_params

struct	matrix_params< architecture::intel_cpu_spr, Ta, Tb, Tc, Td, 0, 0, 0, typename std::enable_if<(!std::is_same_v< Ta, void > &&!std::is_same_v< Tb, void > &&!std::is_same_v< Tc, void >)>::type >

struct	matrix_params< architecture::intel_cpu_spr, Ta, Tb, Tc, Td, sM, sN, sK, typename std::enable_if<(!std::is_same_v< Ta, void > &&!std::is_same_v< Tb, void > &&!std::is_same_v< Tc, void > &&sM !=0 &&sN !=0 &&sK !=0)>::type >

struct	matrix_params< architecture::intel_gpu_dg2_g10, Ta, Tb, Tc, Td, 0, 0, 0, typename std::enable_if<(!std::is_same_v< Ta, void > &&!std::is_same_v< Tb, void > &&!std::is_same_v< Tc, void >)>::type >

struct	matrix_params< architecture::intel_gpu_dg2_g10, Ta, Tb, Tc, Td, sM, sN, sK, typename std::enable_if<(!std::is_same_v< Ta, void > &&!std::is_same_v< Tb, void > &&!std::is_same_v< Tc, void > &&sM !=0 &&sN !=0 &&sK !=0)>::type >

struct	matrix_params< architecture::intel_gpu_dg2_g11, Ta, Tb, Tc, Td, 0, 0, 0, typename std::enable_if<(!std::is_same_v< Ta, void > &&!std::is_same_v< Tb, void > &&!std::is_same_v< Tc, void >)>::type >

struct	matrix_params< architecture::intel_gpu_dg2_g11, Ta, Tb, Tc, Td, sM, sN, sK, typename std::enable_if<(!std::is_same_v< Ta, void > &&!std::is_same_v< Tb, void > &&!std::is_same_v< Tc, void > &&sM !=0 &&sN !=0 &&sK !=0)>::type >

struct	matrix_params< architecture::intel_gpu_dg2_g12, Ta, Tb, Tc, Td, 0, 0, 0, typename std::enable_if<(!std::is_same_v< Ta, void > &&!std::is_same_v< Tb, void > &&!std::is_same_v< Tc, void >)>::type >

struct	matrix_params< architecture::intel_gpu_dg2_g12, Ta, Tb, Tc, Td, sM, sN, sK, typename std::enable_if<(!std::is_same_v< Ta, void > &&!std::is_same_v< Tb, void > &&!std::is_same_v< Tc, void > &&sM !=0 &&sN !=0 &&sK !=0)>::type >

struct	matrix_params< architecture::intel_gpu_pvc, Ta, Tb, Tc, Td, 0, 0, 0, typename std::enable_if<(!std::is_same_v< Ta, void > &&!std::is_same_v< Tb, void > &&!std::is_same_v< Tc, void >)>::type >

struct	matrix_params< architecture::intel_gpu_pvc, Ta, Tb, Tc, Td, sM, sN, sK, typename std::enable_if<(!std::is_same_v< Ta, void > &&!std::is_same_v< Tb, void > &&!std::is_same_v< Tc, void > &&sM !=0 &&sN !=0 &&sK !=0)>::type >

struct	spv_matrix_layout_traits

struct	spv_matrix_use_traits

struct	spv_scope_traits

struct	spv_scope_traits< sycl::group< D > >

struct	spv_scope_traits< sycl::sub_group >

Enumerations
enum	use { use::a, use::b, use::accumulator }

enum	layout { layout::row_major = 0, layout::col_major = 1, layout::ext_intel_packed = 2, layout::dynamic = 3 }

enum	matrix_type { matrix_type::bf16, matrix_type::fp16, matrix_type::tf32, matrix_type::fp32, matrix_type::fp64, matrix_type::sint8, matrix_type::sint16, matrix_type::sint32, matrix_type::sint64, matrix_type::uint8, matrix_type::uint16, matrix_type::uint32, matrix_type::uint64 }

Functions
template<typename Group , typename T , use Use, size_t M, size_t N, layout Layout, typename F >
__SYCL_ALWAYS_INLINE void	joint_matrix_apply (Group sg, joint_matrix< Group, T, Use, M, N, Layout > &jm, F &&lambda)

template<typename Group , typename T , size_t NumRows, size_t NumCols, use Use, layout Layout, typename T2 >
__SYCL_ALWAYS_INLINE void	joint_matrix_fill (Group, joint_matrix< Group, T, Use, NumRows, NumCols, Layout > &res, const T2 &v)

template<typename Group , typename S , typename T , size_t NumRows, size_t NumCols, access::address_space Space, access::decorated IsDecorated, std::enable_if_t< std::is_same< S, std::remove_const_t< T >>::value, bool > = true>
__SYCL_ALWAYS_INLINE void	joint_matrix_load (Group &sg, joint_matrix< Group, S, use::accumulator, NumRows, NumCols, sycl::ext::oneapi::experimental::matrix::layout::dynamic > &res, multi_ptr< T, Space, IsDecorated > src, size_t stride, sycl::ext::oneapi::experimental::matrix::layout Layout)

template<typename Group , typename S , typename T , use Use, size_t NumRows, size_t NumCols, matrix::layout Layout, access::address_space Space, access::decorated IsDecorated, std::enable_if_t< std::is_same< S, std::remove_const_t< T >>::value\|\|(std::is_same< S, precision::tf32 >::value &&std::is_same< std::remove_const_t< T >, float >::value), bool > = true>
__SYCL_ALWAYS_INLINE void	joint_matrix_load (Group &sg, joint_matrix< Group, S, Use, NumRows, NumCols, Layout > &res, multi_ptr< T, Space, IsDecorated > src, size_t stride)

template<typename Group , typename T , size_t NumRows, size_t NumCols, access::address_space Space, access::decorated IsDecorated>
__SYCL_ALWAYS_INLINE void	joint_matrix_store (Group &sg, const joint_matrix< Group, T, use::accumulator, NumRows, NumCols, sycl::ext::oneapi::experimental::matrix::layout::dynamic > &src, multi_ptr< T, Space, IsDecorated > dst, size_t stride, sycl::ext::oneapi::experimental::matrix::layout Layout)

template<typename Group , typename Ta , typename Tb , typename Tc , typename Td , std::size_t M, std::size_t K, std::size_t N, layout LayoutA, layout LayoutB>
__SYCL_ALWAYS_INLINE void	joint_matrix_mad (Group, joint_matrix< Group, Td, use::accumulator, M, N, sycl::ext::oneapi::experimental::matrix::layout::dynamic > &D, const joint_matrix< Group, Ta, use::a, M, K, LayoutA > &A, const joint_matrix< Group, Tb, use::b, K, N, LayoutB > &B, const joint_matrix< Group, Tc, use::accumulator, M, N, sycl::ext::oneapi::experimental::matrix::layout::dynamic > &C)

template<typename Group , typename T1 , typename T2 , size_t Rows, size_t Cols, use Use1, use Use2, layout Layout1, layout Layout2>
void	joint_matrix_copy (Group sg, joint_matrix< Group, T1, Use1, Rows, Cols, Layout1 > &src, joint_matrix< Group, T2, Use2, Rows, Cols, Layout2 > &dst)

__SYCL_ALWAYS_INLINE float	round_to_tf32 (const float &a)

template<typename Ta , typename Tb , typename Tc >
constexpr bool	is_combination_valid_amx (size_t sM, size_t sN, size_t sK)

template<typename Ta , typename Tb , typename Tc >
constexpr bool	are_types_valid_amx ()

template<typename Ta , typename Tb , typename Tc >
constexpr bool	is_combination_valid_xmx8 (size_t sM, size_t sN, size_t sK)

template<typename Ta , typename Tb , typename Tc >
constexpr bool	are_types_valid_xmx8 ()

template<typename Ta , typename Tb , typename Tc >
constexpr bool	is_combination_valid_xmx16 (size_t sM, size_t sN, size_t sK)

template<typename Ta , typename Tb , typename Tc >
constexpr bool	are_types_valid_xmx16 ()

Enumeration Type Documentation

◆ layout

enum sycl::_V1::ext::oneapi::experimental::matrix::layout

strong

Enumerator
row_major
col_major
ext_intel_packed
dynamic

Definition at line 19 of file matrix-unified-utils.hpp.

◆ matrix_type

enum sycl::_V1::ext::oneapi::experimental::matrix::matrix_type

strong

Enumerator
bf16
fp16
tf32
fp32
fp64
sint8
sint16
sint32
sint64
uint8
uint16
uint32
uint64

Definition at line 18 of file query-types.hpp.

◆ use

enum sycl::_V1::ext::oneapi::experimental::matrix::use

strong

Enumerator
a
b
accumulator

Definition at line 17 of file matrix-unified-utils.hpp.