DPC++ Runtime
Runtime libraries for oneAPI DPC++
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
 

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.