matrix-tensorcores.hpp File Reference

#include "matrix-unified-utils.hpp"
#include <sycl/aliases.hpp>
#include <sycl/ext/oneapi/bfloat16.hpp>
#include <sycl/half_type.hpp>
#include <sycl/marray.hpp>
#include <stddef.h>
#include <stdint.h>
#include <type_traits>

Include dependency graph for matrix-tensorcores.hpp:

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

Go to the source code of this file.

Namespaces
	sycl
	sycl::_V1
	sycl::_V1::ext
	sycl::_V1::ext::oneapi
	sycl::_V1::ext::oneapi::experimental
	sycl::_V1::ext::oneapi::experimental::matrix
	sycl::_V1::ext::oneapi::detail
	Objects of the accessor_property_list class are containers for the SYCL properties.

Macros
#define	__SYCL_JOINT_MATRIX_OVERLOAD_ARR(TYPE, USE, M, N, SIZE)
#define	__SYCL_JOINT_MATRIX_OVERLOAD_ARR_ACC(TYPE, M, N, SIZE)
#define	__SYCL_JOINT_MATRIX_OVERLOAD_ARR_PRECISION(PRECISION, USE, M, N, TYPE, SIZE)

Macro Definition Documentation

__SYCL_JOINT_MATRIX_OVERLOAD_ARR

#define __SYCL_JOINT_MATRIX_OVERLOAD_ARR	(		TYPE,
			USE,
			M,
			N,
			SIZE
	)

Value:

  template <sycl::ext::oneapi::experimental::matrix::layout Layout>            \
  struct joint_matrix_cuda<                                                    \
      TYPE, sycl::ext::oneapi::experimental::matrix::use::USE, M, N, Layout,   \
      typename std::enable_if_t<                                               \
          Layout ==                                                            \
              sycl::ext::oneapi::experimental::matrix::layout::row_major ||    \
          Layout ==                                                            \
              sycl::ext::oneapi::experimental::matrix::layout::col_major>> {   \
    marray<TYPE, SIZE> wi_marray;                                              \
  };

Definition at line 46 of file matrix-tensorcores.hpp.

__SYCL_JOINT_MATRIX_OVERLOAD_ARR_ACC

#define __SYCL_JOINT_MATRIX_OVERLOAD_ARR_ACC	(		TYPE,
			M,
			N,
			SIZE
	)

Value:

  template <>                                                                  \
  struct joint_matrix_cuda<                                                    \
      TYPE, sycl::ext::oneapi::experimental::matrix::use::accumulator, M, N,   \
      sycl::ext::oneapi::experimental::matrix::layout::dynamic> {              \
    marray<TYPE, SIZE> wi_marray;                                              \
  };

Definition at line 94 of file matrix-tensorcores.hpp.

__SYCL_JOINT_MATRIX_OVERLOAD_ARR_PRECISION

#define __SYCL_JOINT_MATRIX_OVERLOAD_ARR_PRECISION	(		PRECISION,
			USE,
			M,
			N,
			TYPE,
			SIZE
	)

Value:

  template <sycl::ext::oneapi::experimental::matrix::layout Layout>            \
  struct joint_matrix_cuda<                                                    \
      PRECISION, sycl::ext::oneapi::experimental::matrix::use::USE, M, N,      \
      Layout,                                                                  \
      typename std::enable_if_t<                                               \
          Layout ==                                                            \
              sycl::ext::oneapi::experimental::matrix::layout::row_major ||    \
          Layout ==                                                            \
              sycl::ext::oneapi::experimental::matrix::layout::col_major>> {   \
    marray<TYPE, SIZE> wi_marray;                                              \
  };

Definition at line 115 of file matrix-tensorcores.hpp.