math_mma.hpp File Reference

C++ API. More...

#include "common/core/common.hpp"

Include dependency graph for math_mma.hpp:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Namespaces
namespace	gpu
namespace	gpu::xetla
namespace	gpu::xetla::detail

Functions
template<typename dtype >
constexpr gpu::xetla::argument_type	gpu::xetla::detail::mma_argument_type ()
	convert normal data type to dpas argument type
template<>
constexpr gpu::xetla::argument_type	gpu::xetla::detail::mma_argument_type< tf32 > ()
template<>
constexpr gpu::xetla::argument_type	gpu::xetla::detail::mma_argument_type< float > ()
template<>
constexpr gpu::xetla::argument_type	gpu::xetla::detail::mma_argument_type< int8_t > ()
template<>
constexpr gpu::xetla::argument_type	gpu::xetla::detail::mma_argument_type< uint8_t > ()
template<>
constexpr gpu::xetla::argument_type	gpu::xetla::detail::mma_argument_type< bf16 > ()
template<>
constexpr gpu::xetla::argument_type	gpu::xetla::detail::mma_argument_type< fp16 > ()
template<gpu::xetla::argument_type arg_type>
constexpr __ESIMD_NS::xmx::dpas_argument_type	gpu::xetla::detail::get_argument_type ()
	lookup table for dpas argument type
template<argument_type src1_precision, argument_type src2_precision, int systolic_depth, int repeat_count, typename T , typename T1 , typename T2 , int N, int N1, int N2, typename Sat = xetla_saturation_off_tag>
__XETLA_API xetla_vector< T, N >	gpu::xetla::xetla_mma (xetla_vector< T, N > src0, xetla_vector< T1, N1 > src1, xetla_vector< T2, N2 > src2, Sat sat={})
	description of xetla mma perform matrix multiply add operation

Detailed Description

C++ API.