/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/docs-6.4.3/include/ck/tensor_operation/gpu/grid/gridwise_gemm_waveletmodel.hpp Source File#
gridwise_gemm_waveletmodel.hpp
Go to the documentation of this file.
Definition: ck.hpp:264
static __device__ void RunLoadWavePipeline(const AGridDesc &a_grid_desc, const ABlockDesc &a_block_desc, ABlockTransfer &a_blockwise_copy, const AGridBuffer &a_grid_buf, ABlockBuffer &a_block_buf, const ABlockTransferStep &a_block_copy_step, const BGridDesc &b_grid_desc, const BBlockDesc &b_block_desc, BBlockTransfer &b_blockwise_copy, const BGridBuffer &b_grid_buf, BBlockBuffer &b_block_buf, const BBlockTransferStep &b_block_copy_step, index_t num_loop)
Definition: gridwise_gemm_waveletmodel.hpp:41
__host__ static constexpr __device__ bool IsSupported(index_t)
Definition: gridwise_gemm_waveletmodel.hpp:17
__host__ static constexpr __device__ bool CalculateHasMainLoop(index_t num_loop)
Definition: gridwise_gemm_waveletmodel.hpp:23
Definition: gridwise_gemm_waveletmodel.hpp:11
static __device__ void RunMathWavePipeline(ABlockBuffer &a_block_buf, BBlockBuffer &b_block_buf, const BlockwiseGemm &block_gemm, CThreadBuffer &c_thread_buf, index_t num_loop)
Definition: gridwise_gemm_waveletmodel.hpp:121
__host__ static constexpr __device__ bool IsSupported(index_t)
Definition: gridwise_gemm_waveletmodel.hpp:109
__host__ static constexpr __device__ bool CalculateHasMainLoop(index_t num_loop)
Definition: gridwise_gemm_waveletmodel.hpp:111
Definition: gridwise_gemm_waveletmodel.hpp:103