GridwiseGemmMathWave< TileMathThreadGroup, 1 > Struct Template Reference

GridwiseGemmMathWave&lt; TileMathThreadGroup, 1 &gt; Struct Template Reference#

Composable Kernel: ck::GridwiseGemmMathWave< TileMathThreadGroup, 1 > Struct Template Reference
Static Public Member Functions | List of all members
ck::GridwiseGemmMathWave< TileMathThreadGroup, 1 > Struct Template Reference

#include <gridwise_gemm_waveletmodel.hpp>

Static Public Member Functions

__host__ static __device__ constexpr bool IsSupported (index_t)
__host__ static __device__ constexpr bool CalculateHasMainLoop (index_t num_loop)
template<bool HasMainLoop, typename ABlockBuffer, typename BBlockBuffer, typename BlockwiseGemm, typename CThreadBuffer>
static __device__ void RunMathWavePipeline (ABlockBuffer &a_block_buf, BBlockBuffer &b_block_buf, const BlockwiseGemm &block_gemm, CThreadBuffer &c_thread_buf, index_t num_loop)

Member Function Documentation

◆ CalculateHasMainLoop()

template<typename TileMathThreadGroup>
__host__ static __device__ constexpr bool ck::GridwiseGemmMathWave< TileMathThreadGroup, 1 >::CalculateHasMainLoop ( index_t num_loop)
inlinestaticconstexpr

◆ IsSupported()

template<typename TileMathThreadGroup>
__host__ static __device__ constexpr bool ck::GridwiseGemmMathWave< TileMathThreadGroup, 1 >::IsSupported ( index_t )
inlinestaticconstexpr

◆ RunMathWavePipeline()

template<typename TileMathThreadGroup>
template<bool HasMainLoop, typename ABlockBuffer, typename BBlockBuffer, typename BlockwiseGemm, typename CThreadBuffer>
__device__ void ck::GridwiseGemmMathWave< TileMathThreadGroup, 1 >::RunMathWavePipeline ( ABlockBuffer & a_block_buf,
BBlockBuffer & b_block_buf,
const BlockwiseGemm & block_gemm,
CThreadBuffer & c_thread_buf,
index_t num_loop )
inlinestatic
The documentation for this struct was generated from the following file: