|
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 () |
|