GeneralVectorTensor is a template class that provides Tensor -vector contraction operation, which is a special case of Tensor Tensor contraction. More...

#include <TensorContractionSycl.h>

Public Types
typedef Eigen::TensorSycl::internal::Vectorise< OutScalar, Eigen::SyclDevice, Vectorizable >::PacketReturnType	PacketReturnType

typedef cl::sycl::accessor< OutScalar, 1, cl::sycl::access::mode::read_write, cl::sycl::access::target::local >	Scratch

typedef BlockProperties< is_lhs_vec ? false :true, is_lhs_vec ? false :true, Vectorizable, PacketReturnType >	VecBlockProperties

Public Member Functions
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE	GeneralVectorTensor (Scratch scratch_, const VectorMapper vec_, const TensorMapper mat_, OutAccessor out_res_, const StorageIndex nonContractGroupSize_, const StorageIndex nonContractDim_, const StorageIndex contractDim_)

EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void	operator() (cl::sycl::nd_item< 1 > itemID) const

Static Public Member Functions
template<bool is_internal_block, typename OutPtr >
static EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void	compute_panel (const cl::sycl::nd_item< 1 > &itemID, const VectorMapper &vec, const TensorMapper &mat, OutScalar *local_output, OutPtr out_ptr, const StorageIndex nonContractGroupOffset, const StorageIndex linearLocalThreadId, StorageIndex contractDim, StorageIndex nonContractDim, StorageIndex contractId, StorageIndex nonContractId, StorageIndex globalContractDimOffset, StorageIndex globalNonContractDimOffset, StorageIndex outScratchIndex)

template<typename InputBlockProperties , bool is_internal_block, int CFactor, int GroupSize, typename Input , typename Local >
static EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void	extract_block (const Input &inpt, Local *local_ptr, const StorageIndex &linearLocalThreadId, const StorageIndex &cOffset, const StorageIndex &C)

Public Attributes
Scratch	scratch

const VectorMapper	vec

const TensorMapper	mat

OutAccessor	out_res

const StorageIndex	nonContractGroupSize

const StorageIndex	nonContractDim

const StorageIndex	contractDim

Static Public Attributes
static EIGEN_CONSTEXPR int	PacketSize

static EIGEN_CONSTEXPR StorageIndex	OutScratchOffset

Detailed Description

template<typename OutScalar, typename OutAccessor, typename VectorMapper, typename TensorMapper, typename StorageIndex, typename Properties, StorageIndex KFactor, bool Vectorizable, bool is_lhs_vec, bool IsFinal>
struct Eigen::TensorSycl::internal::GeneralVectorTensor< OutScalar, OutAccessor, VectorMapper, TensorMapper, StorageIndex, Properties, KFactor, Vectorizable, is_lhs_vec, IsFinal >

GeneralVectorTensor is a template class that provides Tensor -vector contraction operation, which is a special case of Tensor Tensor contraction.

Template Parameters

OutScalar	determines the output scalar type
OutAccessor	determines the sycl accessor type for out put (please see the sycl-1.2.1 specification (https://www.khronos.org/registry/SYCL/specs/sycl-1.2.1.pdf) for accessor definition)
VectorMapper	determines the tensor contraction mapper for the vector input (can be lhs or rhs)
TensorMapper	determines the tensor contraction mapper for the tensor input (can be lhs or rhs)
StorageIndex	determines the StorageIndex Type
Properties	determines the Contraction Panel properties
KFactor	determines the number of elements in K dimension in a Tile
Vectorizable	determines whether or not the vectorization is enabled for the Eigen expression.
is_lhs_vec	determines whether lhs is a vector or rhs is a vector
IsFinal	determine if this is the final kernel. If so, the result will be written in a final output. Otherwise, the result of contraction will be written iin a temporary buffer.

Parameters

scratch	determines the local memory containing the vector block for each work-group
vec	determines the vector input (tensor mapper)
mat	determines the tensor input (tensor mapper)
out_res	determines the output vector containing the contraction result
nonContractGroupSize	a logical number determining the number of work-group for non-contracting dimension
nonContractDim	determines the size of non contracting dimension for the flattened tensor
contractDim	determines the size of non contracting dimension for the flattened tensor

Member Typedef Documentation

◆ PacketReturnType

template<typename OutScalar , typename OutAccessor , typename VectorMapper , typename TensorMapper , typename StorageIndex , typename Properties , StorageIndex KFactor, bool Vectorizable, bool is_lhs_vec, bool IsFinal>

typedef Eigen::TensorSycl::internal::Vectorise<OutScalar, Eigen::SyclDevice, Vectorizable>::PacketReturnType Eigen::TensorSycl::internal::GeneralVectorTensor< OutScalar, OutAccessor, VectorMapper, TensorMapper, StorageIndex, Properties, KFactor, Vectorizable, is_lhs_vec, IsFinal >::PacketReturnType

◆ Scratch

template<typename OutScalar , typename OutAccessor , typename VectorMapper , typename TensorMapper , typename StorageIndex , typename Properties , StorageIndex KFactor, bool Vectorizable, bool is_lhs_vec, bool IsFinal>

typedef cl::sycl::accessor<OutScalar, 1, cl::sycl::access::mode::read_write, cl::sycl::access::target::local> Eigen::TensorSycl::internal::GeneralVectorTensor< OutScalar, OutAccessor, VectorMapper, TensorMapper, StorageIndex, Properties, KFactor, Vectorizable, is_lhs_vec, IsFinal >::Scratch

◆ VecBlockProperties

template<typename OutScalar , typename OutAccessor , typename VectorMapper , typename TensorMapper , typename StorageIndex , typename Properties , StorageIndex KFactor, bool Vectorizable, bool is_lhs_vec, bool IsFinal>

typedef BlockProperties<is_lhs_vec ? false : true, is_lhs_vec ? false : true, Vectorizable, PacketReturnType> Eigen::TensorSycl::internal::GeneralVectorTensor< OutScalar, OutAccessor, VectorMapper, TensorMapper, StorageIndex, Properties, KFactor, Vectorizable, is_lhs_vec, IsFinal >::VecBlockProperties

Constructor & Destructor Documentation

◆ GeneralVectorTensor()

template<typename OutScalar , typename OutAccessor , typename VectorMapper , typename TensorMapper , typename StorageIndex , typename Properties , StorageIndex KFactor, bool Vectorizable, bool is_lhs_vec, bool IsFinal>

EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Eigen::TensorSycl::internal::GeneralVectorTensor< OutScalar, OutAccessor, VectorMapper, TensorMapper, StorageIndex, Properties, KFactor, Vectorizable, is_lhs_vec, IsFinal >::GeneralVectorTensor	(	Scratch	scratch_,
		const VectorMapper	vec_,
		const TensorMapper	mat_,
		OutAccessor	out_res_,
		const StorageIndex	nonContractGroupSize_,
		const StorageIndex	nonContractDim_,
		const StorageIndex	contractDim_
	)

inline

       : scratch(scratch_),
         vec(vec_),
         mat(mat_),
         out_res(out_res_),
         nonContractGroupSize(nonContractGroupSize_),
         nonContractDim(nonContractDim_),
         contractDim(contractDim_) {}

Member Function Documentation

◆ compute_panel()

template<typename OutScalar , typename OutAccessor , typename VectorMapper , typename TensorMapper , typename StorageIndex , typename Properties , StorageIndex KFactor, bool Vectorizable, bool is_lhs_vec, bool IsFinal>

template<bool is_internal_block, typename OutPtr >

static EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void Eigen::TensorSycl::internal::GeneralVectorTensor< OutScalar, OutAccessor, VectorMapper, TensorMapper, StorageIndex, Properties, KFactor, Vectorizable, is_lhs_vec, IsFinal >::compute_panel	(	const cl::sycl::nd_item< 1 > &	itemID,
		const VectorMapper &	vec,
		const TensorMapper &	mat,
		OutScalar *	local_output,
		OutPtr	out_ptr,
		const StorageIndex	nonContractGroupOffset,
		const StorageIndex	linearLocalThreadId,
		StorageIndex	contractDim,
		StorageIndex	nonContractDim,
		StorageIndex	contractId,
		StorageIndex	nonContractId,
		StorageIndex	globalContractDimOffset,
		StorageIndex	globalNonContractDimOffset,
		StorageIndex	outScratchIndex
	)

inlinestatic

                                                                                                                    {
     OutScalar outScalar[Properties::WorkLoadPerThreadNC] = {OutScalar(0)};
     // Reading the vector
 #ifdef EIGEN_SYCL_LOCAL_MEM_UNSET_OR_ON
     const StorageIndex vectorOffset = contractGroupOffset + linearLocalThreadId;
     extract_block<VecBlockProperties, is_internal_block, KFactor,
                   Properties::LocalThreadSizeNC * Properties::LocalThreadSizeC>(vec, scratch_ptr, linearLocalThreadId,
                                                                                 vectorOffset, contractDim);
  
     itemID.barrier(cl::sycl::access::fence_space::local_space);
     auto in_scratch_ptr = scratch_ptr + contractId;
 #endif
  
     StorageIndex privateOffsetC = 0;
     EIGEN_UNROLL_LOOP
     for (StorageIndex i = 0; i < Properties::WorkLoadPerThreadC; i++) {
       StorageIndex privateOffsetNC = 0;
       bool contract_conds = ((globalContractDimOffset + privateOffsetC) < contractDim);
 #ifdef EIGEN_SYCL_LOCAL_MEM_UNSET_OR_ON
       auto vecScalar = *in_scratch_ptr;
 #else
       auto vecScalar = (check_boundary<is_internal_block>(contract_conds))
                            ? vec(is_lhs_vec ? StorageIndex(0) : globalContractDimOffset + privateOffsetC,
                                  is_lhs_vec ? globalContractDimOffset + privateOffsetC : StorageIndex(0))
                            : OutScalar(0);
 #endif
       EIGEN_UNROLL_LOOP
       for (StorageIndex j = 0; j < Properties::WorkLoadPerThreadNC; j++) {
         auto matScalar = (check_boundary<is_internal_block>(
                              contract_conds && ((globalNonContractDimOffset + privateOffsetNC) < nonContractDim)))
                              ? mat(is_lhs_vec ? globalContractDimOffset + privateOffsetC
                                               : globalNonContractDimOffset + privateOffsetNC,
                                    is_lhs_vec ? globalNonContractDimOffset + privateOffsetNC
                                               : globalContractDimOffset + privateOffsetC)
                              : OutScalar(0);
  
         outScalar[j] = ::Eigen::internal::pmadd(matScalar, vecScalar, outScalar[j]);
         privateOffsetNC += Properties::LocalThreadSizeNC;
       }
       privateOffsetC += Properties::LocalThreadSizeC;
 #ifdef EIGEN_SYCL_LOCAL_MEM_UNSET_OR_ON
       in_scratch_ptr += Properties::LocalThreadSizeC;
 #endif
     }
  
     auto out_scratch_ptr = local_output + outScratchIndex;
     // Each block of 16*16 element in shared memory should reduce to 16*1
     EIGEN_UNROLL_LOOP
     for (StorageIndex j = 0; j < Properties::WorkLoadPerThreadNC; j++) {
       *out_scratch_ptr = outScalar[j];
  
       out_scratch_ptr += (Properties::LocalThreadSizeNC * Properties::LocalThreadSizeC);
     }
     if (is_lhs_vec) {
       nonContractId = linearLocalThreadId % Properties::LocalThreadSizeNC;
       contractId = linearLocalThreadId / Properties::LocalThreadSizeNC;
       outScratchIndex = nonContractId + contractId * Properties::LocalThreadSizeNC;
     }
  
     out_scratch_ptr = local_output + outScratchIndex;
     EIGEN_UNROLL_LOOP
     for (StorageIndex j = 0; j < Properties::WorkLoadPerThreadNC; j++) {
       EIGEN_UNROLL_LOOP
       for (StorageIndex offset = Properties::LocalThreadSizeC >> 1; offset > 0; offset >>= 1) {
         itemID.barrier(cl::sycl::access::fence_space::local_space);
         if (contractId < offset) {
           StorageIndex myNeigbourId = (Properties::LocalThreadSizeNC * offset);
           *out_scratch_ptr += out_scratch_ptr[myNeigbourId];
         }
       }
       // moving to next 16 by 16 block
       out_scratch_ptr += (Properties::LocalThreadSizeNC * Properties::LocalThreadSizeC);
     }
  
     if (contractId == 0) {
       out_scratch_ptr = local_output + nonContractId;
       StorageIndex global_final_offset = nonContractGroupOffset + nonContractId;
       out_ptr += global_final_offset;
       EIGEN_UNROLL_LOOP
       for (StorageIndex j = 0; j < Properties::WorkLoadPerThreadNC; j++) {
         if (check_boundary<is_internal_block>(global_final_offset < nonContractDim)) {
           auto res = *out_scratch_ptr;
  
           *out_ptr = res;
           out_ptr += Properties::LocalThreadSizeNC;
         }
         // moving to next 16 by 16 block to ge the next 16 reduced elements
         out_scratch_ptr += (Properties::LocalThreadSizeNC * Properties::LocalThreadSizeC);
         if (!(is_internal_block)) global_final_offset += Properties::LocalThreadSizeNC;
       }
     }
   }

References Eigen::TensorSycl::internal::GeneralVectorTensor< OutScalar, OutAccessor, VectorMapper, TensorMapper, StorageIndex, Properties, KFactor, Vectorizable, is_lhs_vec, IsFinal >::contractDim, EIGEN_UNROLL_LOOP, Eigen::TensorSycl::internal::GeneralVectorTensor< OutScalar, OutAccessor, VectorMapper, TensorMapper, StorageIndex, Properties, KFactor, Vectorizable, is_lhs_vec, IsFinal >::extract_block(), i, j, Eigen::TensorSycl::internal::GeneralVectorTensor< OutScalar, OutAccessor, VectorMapper, TensorMapper, StorageIndex, Properties, KFactor, Vectorizable, is_lhs_vec, IsFinal >::mat, Eigen::TensorSycl::internal::GeneralVectorTensor< OutScalar, OutAccessor, VectorMapper, TensorMapper, StorageIndex, Properties, KFactor, Vectorizable, is_lhs_vec, IsFinal >::nonContractDim, Eigen::internal::pmadd(), res, and Eigen::TensorSycl::internal::GeneralVectorTensor< OutScalar, OutAccessor, VectorMapper, TensorMapper, StorageIndex, Properties, KFactor, Vectorizable, is_lhs_vec, IsFinal >::vec.

◆ extract_block()

template<typename OutScalar , typename OutAccessor , typename VectorMapper , typename TensorMapper , typename StorageIndex , typename Properties , StorageIndex KFactor, bool Vectorizable, bool is_lhs_vec, bool IsFinal>

template<typename InputBlockProperties , bool is_internal_block, int CFactor, int GroupSize, typename Input , typename Local >

static EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void Eigen::TensorSycl::internal::GeneralVectorTensor< OutScalar, OutAccessor, VectorMapper, TensorMapper, StorageIndex, Properties, KFactor, Vectorizable, is_lhs_vec, IsFinal >::extract_block	(	const Input &	inpt,
		Local *	local_ptr,
		const StorageIndex &	linearLocalThreadId,
		const StorageIndex &	cOffset,
		const StorageIndex &	C
	)

inlinestatic

                                                                                                                       {
     local_ptr += InputBlockProperties::c_stride * linearLocalThreadId;
     StorageIndex cIndex = cOffset;
     for (StorageIndex cId = 0; cId < CFactor / InputBlockProperties::c_stride; cId++) {
       if (check_boundary<is_internal_block>(cIndex + InputBlockProperties::c_stride - 1 < C)) {
         auto val = read<InputBlockProperties::packet_load, InputBlockProperties::is_coalesced_layout,
                         InputBlockProperties::is_rhs, typename InputBlockProperties::OutType>(inpt, StorageIndex(0),
                                                                                               cIndex, StorageIndex(1));
         write<StorageIndex, 1, data_source::local_mem>(val, local_ptr);
       } else {
         EIGEN_UNROLL_LOOP
         for (StorageIndex i = 0; i < InputBlockProperties::elements_per_access; i++) {
           OutScalar val =
               (cIndex + i < C)
                   ? read<false, InputBlockProperties::is_coalesced_layout, InputBlockProperties::is_rhs, OutScalar>(
                         inpt, StorageIndex(0), cIndex + i, StorageIndex(1))
                   : OutScalar(0);
           write<StorageIndex, 1, data_source::local_mem>(val, local_ptr + i);
         }
       }
       local_ptr += InputBlockProperties::c_stride * GroupSize;
       cIndex += InputBlockProperties::c_stride * GroupSize;
     }
   }

References EIGEN_UNROLL_LOOP, i, Eigen::TensorSycl::internal::read(), and calibrate::val.

Referenced by Eigen::TensorSycl::internal::GeneralVectorTensor< OutScalar, OutAccessor, VectorMapper, TensorMapper, StorageIndex, Properties, KFactor, Vectorizable, is_lhs_vec, IsFinal >::compute_panel().

◆ operator()()

template<typename OutScalar , typename OutAccessor , typename VectorMapper , typename TensorMapper , typename StorageIndex , typename Properties , StorageIndex KFactor, bool Vectorizable, bool is_lhs_vec, bool IsFinal>

EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void Eigen::TensorSycl::internal::GeneralVectorTensor< OutScalar, OutAccessor, VectorMapper, TensorMapper, StorageIndex, Properties, KFactor, Vectorizable, is_lhs_vec, IsFinal >::operator() ( cl::sycl::nd_item< 1 > itemID ) const

inline

                                                                                        {
     auto scratch_ptr = scratch.get_pointer();
     const StorageIndex linearLocalThreadId = itemID.get_local_id(0);
     StorageIndex nonContractId = is_lhs_vec ? linearLocalThreadId / Properties::LocalThreadSizeC
                                             : linearLocalThreadId % Properties::LocalThreadSizeNC;
     StorageIndex contractId = is_lhs_vec ? linearLocalThreadId % Properties::LocalThreadSizeC
                                          : linearLocalThreadId / Properties::LocalThreadSizeNC;
     const StorageIndex cGroupSize = itemID.get_group_range(0) / nonContractGroupSize;
     const StorageIndex nonContractGroupId =
         is_lhs_vec ? itemID.get_group(0) / cGroupSize : itemID.get_group(0) % nonContractGroupSize;
     const StorageIndex contractGroupId =
         is_lhs_vec ? itemID.get_group(0) % cGroupSize : itemID.get_group(0) / nonContractGroupSize;
     auto out_ptr = out_res + (IsFinal ? 0 : contractGroupId * nonContractDim);
  
     const StorageIndex nonContractGroupOffset = nonContractGroupId * Properties::TileSizeDimNC;
     const StorageIndex contractGroupOffset = contractGroupId * Properties::TileSizeDimC;
     auto outScratchIndex = nonContractId + contractId * Properties::LocalThreadSizeNC;
     const StorageIndex globalNonContractDimOffset = nonContractGroupOffset + nonContractId;
     const StorageIndex globalContractDimOffset = contractGroupOffset + contractId;
     auto local_output = scratch_ptr + OutScratchOffset;
     const bool is_internal = nonContractDim - nonContractGroupOffset >= Properties::TileSizeDimNC &&
                              contractDim - contractGroupOffset >= Properties::TileSizeDimC;
     is_internal
         ? compute_panel<true>(itemID, vec, mat, local_output, out_ptr,
 #ifdef EIGEN_SYCL_LOCAL_MEM_UNSET_OR_ON
                               scratch_ptr, contractGroupOffset,
 #endif
                               nonContractGroupOffset, linearLocalThreadId, contractDim, nonContractDim, contractId,
                               nonContractId, globalContractDimOffset, globalNonContractDimOffset, outScratchIndex)
         : compute_panel<false>(itemID, vec, mat, local_output, out_ptr,
 #ifdef EIGEN_SYCL_LOCAL_MEM_UNSET_OR_ON
                                scratch_ptr, contractGroupOffset,
 #endif
                                nonContractGroupOffset, linearLocalThreadId, contractDim, nonContractDim, contractId,
                                nonContractId, globalContractDimOffset, globalNonContractDimOffset, outScratchIndex);
   }

References Eigen::TensorSycl::internal::GeneralVectorTensor< OutScalar, OutAccessor, VectorMapper, TensorMapper, StorageIndex, Properties, KFactor, Vectorizable, is_lhs_vec, IsFinal >::contractDim, EIGEN_SYCL_LOCAL_MEM_UNSET_OR_ON, Eigen::TensorSycl::internal::GeneralVectorTensor< OutScalar, OutAccessor, VectorMapper, TensorMapper, StorageIndex, Properties, KFactor, Vectorizable, is_lhs_vec, IsFinal >::nonContractDim, Eigen::TensorSycl::internal::GeneralVectorTensor< OutScalar, OutAccessor, VectorMapper, TensorMapper, StorageIndex, Properties, KFactor, Vectorizable, is_lhs_vec, IsFinal >::nonContractGroupSize, Eigen::TensorSycl::internal::GeneralVectorTensor< OutScalar, OutAccessor, VectorMapper, TensorMapper, StorageIndex, Properties, KFactor, Vectorizable, is_lhs_vec, IsFinal >::out_res, Eigen::TensorSycl::internal::GeneralVectorTensor< OutScalar, OutAccessor, VectorMapper, TensorMapper, StorageIndex, Properties, KFactor, Vectorizable, is_lhs_vec, IsFinal >::OutScratchOffset, Eigen::TensorSycl::internal::GeneralVectorTensor< OutScalar, OutAccessor, VectorMapper, TensorMapper, StorageIndex, Properties, KFactor, Vectorizable, is_lhs_vec, IsFinal >::scratch, and Eigen::TensorSycl::internal::GeneralVectorTensor< OutScalar, OutAccessor, VectorMapper, TensorMapper, StorageIndex, Properties, KFactor, Vectorizable, is_lhs_vec, IsFinal >::vec.

Member Data Documentation

◆ contractDim

template<typename OutScalar , typename OutAccessor , typename VectorMapper , typename TensorMapper , typename StorageIndex , typename Properties , StorageIndex KFactor, bool Vectorizable, bool is_lhs_vec, bool IsFinal>

const StorageIndex Eigen::TensorSycl::internal::GeneralVectorTensor< OutScalar, OutAccessor, VectorMapper, TensorMapper, StorageIndex, Properties, KFactor, Vectorizable, is_lhs_vec, IsFinal >::contractDim

Referenced by Eigen::TensorSycl::internal::GeneralVectorTensor< OutScalar, OutAccessor, VectorMapper, TensorMapper, StorageIndex, Properties, KFactor, Vectorizable, is_lhs_vec, IsFinal >::compute_panel(), and Eigen::TensorSycl::internal::GeneralVectorTensor< OutScalar, OutAccessor, VectorMapper, TensorMapper, StorageIndex, Properties, KFactor, Vectorizable, is_lhs_vec, IsFinal >::operator()().

◆ mat

template<typename OutScalar , typename OutAccessor , typename VectorMapper , typename TensorMapper , typename StorageIndex , typename Properties , StorageIndex KFactor, bool Vectorizable, bool is_lhs_vec, bool IsFinal>

const TensorMapper Eigen::TensorSycl::internal::GeneralVectorTensor< OutScalar, OutAccessor, VectorMapper, TensorMapper, StorageIndex, Properties, KFactor, Vectorizable, is_lhs_vec, IsFinal >::mat

Referenced by Eigen::TensorSycl::internal::GeneralVectorTensor< OutScalar, OutAccessor, VectorMapper, TensorMapper, StorageIndex, Properties, KFactor, Vectorizable, is_lhs_vec, IsFinal >::compute_panel().

◆ nonContractDim

template<typename OutScalar , typename OutAccessor , typename VectorMapper , typename TensorMapper , typename StorageIndex , typename Properties , StorageIndex KFactor, bool Vectorizable, bool is_lhs_vec, bool IsFinal>

const StorageIndex Eigen::TensorSycl::internal::GeneralVectorTensor< OutScalar, OutAccessor, VectorMapper, TensorMapper, StorageIndex, Properties, KFactor, Vectorizable, is_lhs_vec, IsFinal >::nonContractDim

Referenced by Eigen::TensorSycl::internal::GeneralVectorTensor< OutScalar, OutAccessor, VectorMapper, TensorMapper, StorageIndex, Properties, KFactor, Vectorizable, is_lhs_vec, IsFinal >::compute_panel(), and Eigen::TensorSycl::internal::GeneralVectorTensor< OutScalar, OutAccessor, VectorMapper, TensorMapper, StorageIndex, Properties, KFactor, Vectorizable, is_lhs_vec, IsFinal >::operator()().

◆ nonContractGroupSize

template<typename OutScalar , typename OutAccessor , typename VectorMapper , typename TensorMapper , typename StorageIndex , typename Properties , StorageIndex KFactor, bool Vectorizable, bool is_lhs_vec, bool IsFinal>

const StorageIndex Eigen::TensorSycl::internal::GeneralVectorTensor< OutScalar, OutAccessor, VectorMapper, TensorMapper, StorageIndex, Properties, KFactor, Vectorizable, is_lhs_vec, IsFinal >::nonContractGroupSize

Referenced by Eigen::TensorSycl::internal::GeneralVectorTensor< OutScalar, OutAccessor, VectorMapper, TensorMapper, StorageIndex, Properties, KFactor, Vectorizable, is_lhs_vec, IsFinal >::operator()().

◆ out_res

template<typename OutScalar , typename OutAccessor , typename VectorMapper , typename TensorMapper , typename StorageIndex , typename Properties , StorageIndex KFactor, bool Vectorizable, bool is_lhs_vec, bool IsFinal>

OutAccessor Eigen::TensorSycl::internal::GeneralVectorTensor< OutScalar, OutAccessor, VectorMapper, TensorMapper, StorageIndex, Properties, KFactor, Vectorizable, is_lhs_vec, IsFinal >::out_res

Referenced by Eigen::TensorSycl::internal::GeneralVectorTensor< OutScalar, OutAccessor, VectorMapper, TensorMapper, StorageIndex, Properties, KFactor, Vectorizable, is_lhs_vec, IsFinal >::operator()().

◆ OutScratchOffset

template<typename OutScalar , typename OutAccessor , typename VectorMapper , typename TensorMapper , typename StorageIndex , typename Properties , StorageIndex KFactor, bool Vectorizable, bool is_lhs_vec, bool IsFinal>

EIGEN_CONSTEXPR StorageIndex Eigen::TensorSycl::internal::GeneralVectorTensor< OutScalar, OutAccessor, VectorMapper, TensorMapper, StorageIndex, Properties, KFactor, Vectorizable, is_lhs_vec, IsFinal >::OutScratchOffset

static

Initial value:

=

KFactor * Properties::LocalThreadSizeC * Properties::LocalThreadSizeNC

Referenced by Eigen::TensorSycl::internal::GeneralVectorTensor< OutScalar, OutAccessor, VectorMapper, TensorMapper, StorageIndex, Properties, KFactor, Vectorizable, is_lhs_vec, IsFinal >::operator()().

◆ PacketSize

template<typename OutScalar , typename OutAccessor , typename VectorMapper , typename TensorMapper , typename StorageIndex , typename Properties , StorageIndex KFactor, bool Vectorizable, bool is_lhs_vec, bool IsFinal>

EIGEN_CONSTEXPR int Eigen::TensorSycl::internal::GeneralVectorTensor< OutScalar, OutAccessor, VectorMapper, TensorMapper, StorageIndex, Properties, KFactor, Vectorizable, is_lhs_vec, IsFinal >::PacketSize

static

Initial value:

=

Eigen::TensorSycl::internal::Vectorise<OutScalar, Eigen::SyclDevice, Vectorizable>::PacketSize

◆ scratch

template<typename OutScalar , typename OutAccessor , typename VectorMapper , typename TensorMapper , typename StorageIndex , typename Properties , StorageIndex KFactor, bool Vectorizable, bool is_lhs_vec, bool IsFinal>

Scratch Eigen::TensorSycl::internal::GeneralVectorTensor< OutScalar, OutAccessor, VectorMapper, TensorMapper, StorageIndex, Properties, KFactor, Vectorizable, is_lhs_vec, IsFinal >::scratch

Referenced by Eigen::TensorSycl::internal::GeneralVectorTensor< OutScalar, OutAccessor, VectorMapper, TensorMapper, StorageIndex, Properties, KFactor, Vectorizable, is_lhs_vec, IsFinal >::operator()().

◆ vec

template<typename OutScalar , typename OutAccessor , typename VectorMapper , typename TensorMapper , typename StorageIndex , typename Properties , StorageIndex KFactor, bool Vectorizable, bool is_lhs_vec, bool IsFinal>

const VectorMapper Eigen::TensorSycl::internal::GeneralVectorTensor< OutScalar, OutAccessor, VectorMapper, TensorMapper, StorageIndex, Properties, KFactor, Vectorizable, is_lhs_vec, IsFinal >::vec

Referenced by Eigen::TensorSycl::internal::GeneralVectorTensor< OutScalar, OutAccessor, VectorMapper, TensorMapper, StorageIndex, Properties, KFactor, Vectorizable, is_lhs_vec, IsFinal >::compute_panel(), and Eigen::TensorSycl::internal::GeneralVectorTensor< OutScalar, OutAccessor, VectorMapper, TensorMapper, StorageIndex, Properties, KFactor, Vectorizable, is_lhs_vec, IsFinal >::operator()().

The documentation for this struct was generated from the following file:

TensorContractionSycl.h

Public Types

Public Member Functions

Static Public Member Functions

Public Attributes

Static Public Attributes

Detailed Description

Member Typedef Documentation

◆ PacketReturnType

◆ Scratch

◆ VecBlockProperties

Constructor & Destructor Documentation

◆ GeneralVectorTensor()

Member Function Documentation

◆ compute_panel()

◆ extract_block()

◆ operator()()

Member Data Documentation

◆ contractDim

◆ mat

◆ nonContractDim

◆ nonContractGroupSize

◆ out_res

◆ OutScratchOffset

◆ PacketSize

◆ scratch

◆ vec