batchlas::sycl_gemm::Gemm128x128Body< T, AlignedFastPath, StagedEpilogue > Struct Template Reference

Overview

template<typename T, bool AlignedFastPath, bool StagedEpilogue>
struct batchlas::sycl_gemm::Gemm128x128Body< T, AlignedFastPath, StagedEpilogue >

Definition at line 127 of file register_128x128.hh.

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

Functions

◆ operator()()

template<typename T, bool AlignedFastPath, bool StagedEpilogue>
void batchlas::sycl_gemm::Gemm128x128Body< T, AlignedFastPath, StagedEpilogue >::operator() ( sycl::nd_item< 3 > item) const
inline

Definition at line 147 of file register_128x128.hh.

Variables

◆ TileM

template<typename T, bool AlignedFastPath, bool StagedEpilogue>
int batchlas::sycl_gemm::Gemm128x128Body< T, AlignedFastPath, StagedEpilogue >::TileM = 128
staticconstexpr

Definition at line 128 of file register_128x128.hh.

◆ TileN

template<typename T, bool AlignedFastPath, bool StagedEpilogue>
int batchlas::sycl_gemm::Gemm128x128Body< T, AlignedFastPath, StagedEpilogue >::TileN = 128
static

Definition at line 128 of file register_128x128.hh.

◆ TileK

template<typename T, bool AlignedFastPath, bool StagedEpilogue>
int batchlas::sycl_gemm::Gemm128x128Body< T, AlignedFastPath, StagedEpilogue >::TileK = 8
static

Definition at line 128 of file register_128x128.hh.

◆ ThreadTile

template<typename T, bool AlignedFastPath, bool StagedEpilogue>
int batchlas::sycl_gemm::Gemm128x128Body< T, AlignedFastPath, StagedEpilogue >::ThreadTile = 8
staticconstexpr

Definition at line 129 of file register_128x128.hh.

◆ Band

template<typename T, bool AlignedFastPath, bool StagedEpilogue>
int batchlas::sycl_gemm::Gemm128x128Body< T, AlignedFastPath, StagedEpilogue >::Band = 4
static

Definition at line 129 of file register_128x128.hh.

◆ Threads

template<typename T, bool AlignedFastPath, bool StagedEpilogue>
int batchlas::sycl_gemm::Gemm128x128Body< T, AlignedFastPath, StagedEpilogue >::Threads = 256
static

Definition at line 129 of file register_128x128.hh.

◆ BandM

template<typename T, bool AlignedFastPath, bool StagedEpilogue>
int batchlas::sycl_gemm::Gemm128x128Body< T, AlignedFastPath, StagedEpilogue >::BandM = 32
staticconstexpr

Definition at line 130 of file register_128x128.hh.

◆ BandN

template<typename T, bool AlignedFastPath, bool StagedEpilogue>
int batchlas::sycl_gemm::Gemm128x128Body< T, AlignedFastPath, StagedEpilogue >::BandN = 16
static

Definition at line 130 of file register_128x128.hh.

◆ AStride

template<typename T, bool AlignedFastPath, bool StagedEpilogue>
int batchlas::sycl_gemm::Gemm128x128Body< T, AlignedFastPath, StagedEpilogue >::AStride = TileM
staticconstexpr

Definition at line 131 of file register_128x128.hh.

◆ BStride

template<typename T, bool AlignedFastPath, bool StagedEpilogue>
int batchlas::sycl_gemm::Gemm128x128Body< T, AlignedFastPath, StagedEpilogue >::BStride = TileN
static

Definition at line 131 of file register_128x128.hh.

◆ kMinGroupsPerCu

template<typename T, bool AlignedFastPath, bool StagedEpilogue>
int batchlas::sycl_gemm::Gemm128x128Body< T, AlignedFastPath, StagedEpilogue >::kMinGroupsPerCu = 2
staticconstexpr

Definition at line 132 of file register_128x128.hh.

◆ StageStride

template<typename T, bool AlignedFastPath, bool StagedEpilogue>
int batchlas::sycl_gemm::Gemm128x128Body< T, AlignedFastPath, StagedEpilogue >::StageStride = TileM + 4
staticconstexpr

Definition at line 133 of file register_128x128.hh.

◆ TileAElems

template<typename T, bool AlignedFastPath, bool StagedEpilogue>
int batchlas::sycl_gemm::Gemm128x128Body< T, AlignedFastPath, StagedEpilogue >::TileAElems = StagedEpilogue ? 32 * StageStride : 2 * TileK * AStride
staticconstexpr

Definition at line 134 of file register_128x128.hh.

◆ a_ptr

template<typename T, bool AlignedFastPath, bool StagedEpilogue>
const T* batchlas::sycl_gemm::Gemm128x128Body< T, AlignedFastPath, StagedEpilogue >::a_ptr

Definition at line 137 of file register_128x128.hh.

◆ b_ptr

template<typename T, bool AlignedFastPath, bool StagedEpilogue>
const T* batchlas::sycl_gemm::Gemm128x128Body< T, AlignedFastPath, StagedEpilogue >::b_ptr

Definition at line 138 of file register_128x128.hh.

◆ c_ptr

template<typename T, bool AlignedFastPath, bool StagedEpilogue>
T* batchlas::sycl_gemm::Gemm128x128Body< T, AlignedFastPath, StagedEpilogue >::c_ptr

Definition at line 139 of file register_128x128.hh.

◆ m

template<typename T, bool AlignedFastPath, bool StagedEpilogue>
int batchlas::sycl_gemm::Gemm128x128Body< T, AlignedFastPath, StagedEpilogue >::m

Definition at line 140 of file register_128x128.hh.

◆ n

template<typename T, bool AlignedFastPath, bool StagedEpilogue>
int batchlas::sycl_gemm::Gemm128x128Body< T, AlignedFastPath, StagedEpilogue >::n

Definition at line 140 of file register_128x128.hh.

◆ k

template<typename T, bool AlignedFastPath, bool StagedEpilogue>
int batchlas::sycl_gemm::Gemm128x128Body< T, AlignedFastPath, StagedEpilogue >::k

Definition at line 140 of file register_128x128.hh.

◆ lda

template<typename T, bool AlignedFastPath, bool StagedEpilogue>
int batchlas::sycl_gemm::Gemm128x128Body< T, AlignedFastPath, StagedEpilogue >::lda

Definition at line 140 of file register_128x128.hh.

◆ ldb

template<typename T, bool AlignedFastPath, bool StagedEpilogue>
int batchlas::sycl_gemm::Gemm128x128Body< T, AlignedFastPath, StagedEpilogue >::ldb

Definition at line 140 of file register_128x128.hh.

◆ ldc

template<typename T, bool AlignedFastPath, bool StagedEpilogue>
int batchlas::sycl_gemm::Gemm128x128Body< T, AlignedFastPath, StagedEpilogue >::ldc

Definition at line 140 of file register_128x128.hh.

◆ stride_a

template<typename T, bool AlignedFastPath, bool StagedEpilogue>
std::ptrdiff_t batchlas::sycl_gemm::Gemm128x128Body< T, AlignedFastPath, StagedEpilogue >::stride_a

Definition at line 141 of file register_128x128.hh.

◆ stride_b

template<typename T, bool AlignedFastPath, bool StagedEpilogue>
std::ptrdiff_t batchlas::sycl_gemm::Gemm128x128Body< T, AlignedFastPath, StagedEpilogue >::stride_b

Definition at line 141 of file register_128x128.hh.

◆ stride_c

template<typename T, bool AlignedFastPath, bool StagedEpilogue>
std::ptrdiff_t batchlas::sycl_gemm::Gemm128x128Body< T, AlignedFastPath, StagedEpilogue >::stride_c

Definition at line 141 of file register_128x128.hh.

◆ batch

template<typename T, bool AlignedFastPath, bool StagedEpilogue>
int batchlas::sycl_gemm::Gemm128x128Body< T, AlignedFastPath, StagedEpilogue >::batch

Definition at line 142 of file register_128x128.hh.

◆ alpha

template<typename T, bool AlignedFastPath, bool StagedEpilogue>
T batchlas::sycl_gemm::Gemm128x128Body< T, AlignedFastPath, StagedEpilogue >::alpha

Definition at line 143 of file register_128x128.hh.

◆ beta

template<typename T, bool AlignedFastPath, bool StagedEpilogue>
T batchlas::sycl_gemm::Gemm128x128Body< T, AlignedFastPath, StagedEpilogue >::beta

Definition at line 143 of file register_128x128.hh.

◆ tile_a

template<typename T, bool AlignedFastPath, bool StagedEpilogue>
sycl::local_accessor<T, 1> batchlas::sycl_gemm::Gemm128x128Body< T, AlignedFastPath, StagedEpilogue >::tile_a

Definition at line 144 of file register_128x128.hh.

◆ tile_b

template<typename T, bool AlignedFastPath, bool StagedEpilogue>
sycl::local_accessor<T, 1> batchlas::sycl_gemm::Gemm128x128Body< T, AlignedFastPath, StagedEpilogue >::tile_b

Definition at line 144 of file register_128x128.hh.