BatchLAS: batchlas::sycl_gemm::RegisterTilePolicy< TileM_, TileN_, TileK_, ThreadTileRows_, ThreadTileCols_, VecA_, VecB_, UnrollK_, Stages_ > Struct Template Reference

Overview

template<int TileM_, int TileN_, int TileK_, int ThreadTileRows_, int ThreadTileCols_ = ThreadTileRows_, int VecA_ = 4, int VecB_ = 4, int UnrollK_ = 1, int Stages_ = 1>
struct batchlas::sycl_gemm::RegisterTilePolicy< TileM_, TileN_, TileK_, ThreadTileRows_, ThreadTileCols_, VecA_, VecB_, UnrollK_, Stages_ >

Definition at line 56 of file register_tiled_common.hh.

#include </home/runner/work/BatchLAS/BatchLAS/src/ops/gemm/kernels/register_tiled_common.hh>

Variables

◆ TileM

template<int TileM_, int TileN_, int TileK_, int ThreadTileRows_, int ThreadTileCols_ = ThreadTileRows_, int VecA_ = 4, int VecB_ = 4, int UnrollK_ = 1, int Stages_ = 1>

int batchlas::sycl_gemm::RegisterTilePolicy< TileM_, TileN_, TileK_, ThreadTileRows_, ThreadTileCols_, VecA_, VecB_, UnrollK_, Stages_ >::TileM = TileM_

staticconstexpr

Definition at line 57 of file register_tiled_common.hh.

◆ TileN

template<int TileM_, int TileN_, int TileK_, int ThreadTileRows_, int ThreadTileCols_ = ThreadTileRows_, int VecA_ = 4, int VecB_ = 4, int UnrollK_ = 1, int Stages_ = 1>

int batchlas::sycl_gemm::RegisterTilePolicy< TileM_, TileN_, TileK_, ThreadTileRows_, ThreadTileCols_, VecA_, VecB_, UnrollK_, Stages_ >::TileN = TileN_

staticconstexpr

Definition at line 58 of file register_tiled_common.hh.

◆ TileK

template<int TileM_, int TileN_, int TileK_, int ThreadTileRows_, int ThreadTileCols_ = ThreadTileRows_, int VecA_ = 4, int VecB_ = 4, int UnrollK_ = 1, int Stages_ = 1>

int batchlas::sycl_gemm::RegisterTilePolicy< TileM_, TileN_, TileK_, ThreadTileRows_, ThreadTileCols_, VecA_, VecB_, UnrollK_, Stages_ >::TileK = TileK_

staticconstexpr

Definition at line 59 of file register_tiled_common.hh.

◆ ThreadTileRows

template<int TileM_, int TileN_, int TileK_, int ThreadTileRows_, int ThreadTileCols_ = ThreadTileRows_, int VecA_ = 4, int VecB_ = 4, int UnrollK_ = 1, int Stages_ = 1>

int batchlas::sycl_gemm::RegisterTilePolicy< TileM_, TileN_, TileK_, ThreadTileRows_, ThreadTileCols_, VecA_, VecB_, UnrollK_, Stages_ >::ThreadTileRows = ThreadTileRows_

staticconstexpr

Definition at line 60 of file register_tiled_common.hh.

◆ ThreadTileCols

template<int TileM_, int TileN_, int TileK_, int ThreadTileRows_, int ThreadTileCols_ = ThreadTileRows_, int VecA_ = 4, int VecB_ = 4, int UnrollK_ = 1, int Stages_ = 1>

int batchlas::sycl_gemm::RegisterTilePolicy< TileM_, TileN_, TileK_, ThreadTileRows_, ThreadTileCols_, VecA_, VecB_, UnrollK_, Stages_ >::ThreadTileCols = ThreadTileCols_

staticconstexpr

Definition at line 61 of file register_tiled_common.hh.

◆ VecA

template<int TileM_, int TileN_, int TileK_, int ThreadTileRows_, int ThreadTileCols_ = ThreadTileRows_, int VecA_ = 4, int VecB_ = 4, int UnrollK_ = 1, int Stages_ = 1>

int batchlas::sycl_gemm::RegisterTilePolicy< TileM_, TileN_, TileK_, ThreadTileRows_, ThreadTileCols_, VecA_, VecB_, UnrollK_, Stages_ >::VecA = VecA_

staticconstexpr

Definition at line 62 of file register_tiled_common.hh.

◆ VecB

template<int TileM_, int TileN_, int TileK_, int ThreadTileRows_, int ThreadTileCols_ = ThreadTileRows_, int VecA_ = 4, int VecB_ = 4, int UnrollK_ = 1, int Stages_ = 1>

int batchlas::sycl_gemm::RegisterTilePolicy< TileM_, TileN_, TileK_, ThreadTileRows_, ThreadTileCols_, VecA_, VecB_, UnrollK_, Stages_ >::VecB = VecB_

staticconstexpr

Definition at line 63 of file register_tiled_common.hh.

◆ UnrollK

template<int TileM_, int TileN_, int TileK_, int ThreadTileRows_, int ThreadTileCols_ = ThreadTileRows_, int VecA_ = 4, int VecB_ = 4, int UnrollK_ = 1, int Stages_ = 1>

int batchlas::sycl_gemm::RegisterTilePolicy< TileM_, TileN_, TileK_, ThreadTileRows_, ThreadTileCols_, VecA_, VecB_, UnrollK_, Stages_ >::UnrollK = UnrollK_

staticconstexpr

Definition at line 64 of file register_tiled_common.hh.

◆ Stages

template<int TileM_, int TileN_, int TileK_, int ThreadTileRows_, int ThreadTileCols_ = ThreadTileRows_, int VecA_ = 4, int VecB_ = 4, int UnrollK_ = 1, int Stages_ = 1>

int batchlas::sycl_gemm::RegisterTilePolicy< TileM_, TileN_, TileK_, ThreadTileRows_, ThreadTileCols_, VecA_, VecB_, UnrollK_, Stages_ >::Stages = Stages_

staticconstexpr

Definition at line 65 of file register_tiled_common.hh.

◆ LocalRows

template<int TileM_, int TileN_, int TileK_, int ThreadTileRows_, int ThreadTileCols_ = ThreadTileRows_, int VecA_ = 4, int VecB_ = 4, int UnrollK_ = 1, int Stages_ = 1>

int batchlas::sycl_gemm::RegisterTilePolicy< TileM_, TileN_, TileK_, ThreadTileRows_, ThreadTileCols_, VecA_, VecB_, UnrollK_, Stages_ >::LocalRows = TileM / ThreadTileRows

staticconstexpr

Definition at line 66 of file register_tiled_common.hh.

◆ LocalCols

template<int TileM_, int TileN_, int TileK_, int ThreadTileRows_, int ThreadTileCols_ = ThreadTileRows_, int VecA_ = 4, int VecB_ = 4, int UnrollK_ = 1, int Stages_ = 1>

int batchlas::sycl_gemm::RegisterTilePolicy< TileM_, TileN_, TileK_, ThreadTileRows_, ThreadTileCols_, VecA_, VecB_, UnrollK_, Stages_ >::LocalCols = TileN / ThreadTileCols

staticconstexpr

Definition at line 67 of file register_tiled_common.hh.

◆ ThreadsPerGroup

template<int TileM_, int TileN_, int TileK_, int ThreadTileRows_, int ThreadTileCols_ = ThreadTileRows_, int VecA_ = 4, int VecB_ = 4, int UnrollK_ = 1, int Stages_ = 1>

int batchlas::sycl_gemm::RegisterTilePolicy< TileM_, TileN_, TileK_, ThreadTileRows_, ThreadTileCols_, VecA_, VecB_, UnrollK_, Stages_ >::ThreadsPerGroup = LocalRows * LocalCols

staticconstexpr

Definition at line 68 of file register_tiled_common.hh.

◆ TileAStride

template<int TileM_, int TileN_, int TileK_, int ThreadTileRows_, int ThreadTileCols_ = ThreadTileRows_, int VecA_ = 4, int VecB_ = 4, int UnrollK_ = 1, int Stages_ = 1>

int batchlas::sycl_gemm::RegisterTilePolicy< TileM_, TileN_, TileK_, ThreadTileRows_, ThreadTileCols_, VecA_, VecB_, UnrollK_, Stages_ >::TileAStride = TileM + 1

staticconstexpr

Definition at line 69 of file register_tiled_common.hh.

◆ TileBStride

template<int TileM_, int TileN_, int TileK_, int ThreadTileRows_, int ThreadTileCols_ = ThreadTileRows_, int VecA_ = 4, int VecB_ = 4, int UnrollK_ = 1, int Stages_ = 1>

int batchlas::sycl_gemm::RegisterTilePolicy< TileM_, TileN_, TileK_, ThreadTileRows_, ThreadTileCols_, VecA_, VecB_, UnrollK_, Stages_ >::TileBStride = TileK + 1

staticconstexpr

Definition at line 70 of file register_tiled_common.hh.

◆ StageASize

template<int TileM_, int TileN_, int TileK_, int ThreadTileRows_, int ThreadTileCols_ = ThreadTileRows_, int VecA_ = 4, int VecB_ = 4, int UnrollK_ = 1, int Stages_ = 1>

int batchlas::sycl_gemm::RegisterTilePolicy< TileM_, TileN_, TileK_, ThreadTileRows_, ThreadTileCols_, VecA_, VecB_, UnrollK_, Stages_ >::StageASize = TileAStride * TileK

staticconstexpr

Definition at line 71 of file register_tiled_common.hh.

◆ StageBSize

template<int TileM_, int TileN_, int TileK_, int ThreadTileRows_, int ThreadTileCols_ = ThreadTileRows_, int VecA_ = 4, int VecB_ = 4, int UnrollK_ = 1, int Stages_ = 1>

int batchlas::sycl_gemm::RegisterTilePolicy< TileM_, TileN_, TileK_, ThreadTileRows_, ThreadTileCols_, VecA_, VecB_, UnrollK_, Stages_ >::StageBSize = TileBStride * TileN

staticconstexpr

Definition at line 72 of file register_tiled_common.hh.