SplitKBatchOffset Struct Reference

SplitKBatchOffset Struct Reference#

Composable Kernel: ck_tile::GroupedConvolutionBackwardWeightKernel< GroupedConvTraitsType_, TilePartitioner_, GemmPipeline_, EpiloguePipeline_ >::SplitKBatchOffset Struct Reference
Public Member Functions | Public Attributes | List of all members
ck_tile::GroupedConvolutionBackwardWeightKernel< GroupedConvTraitsType_, TilePartitioner_, GemmPipeline_, EpiloguePipeline_ >::SplitKBatchOffset Struct Reference

#include <grouped_convolution_backward_weight_kernel.hpp>

Public Member Functions

__device__ SplitKBatchOffset (const GroupedConvBwdWeightKernelArgsSpecialized &kargs, const std::size_t k_id=blockIdx.z)

Public Attributes

index_t a_k_split_offset
index_t b_k_split_offset
index_t splitted_k

Constructor & Destructor Documentation

◆ SplitKBatchOffset()

template<typename GroupedConvTraitsType_, typename TilePartitioner_, typename GemmPipeline_, typename EpiloguePipeline_>
__device__ ck_tile::GroupedConvolutionBackwardWeightKernel< GroupedConvTraitsType_, TilePartitioner_, GemmPipeline_, EpiloguePipeline_ >::SplitKBatchOffset::SplitKBatchOffset ( const GroupedConvBwdWeightKernelArgsSpecialized & kargs,
const std::size_t k_id = blockIdx.z )
inline

Member Data Documentation

◆ a_k_split_offset

template<typename GroupedConvTraitsType_, typename TilePartitioner_, typename GemmPipeline_, typename EpiloguePipeline_>
index_t ck_tile::GroupedConvolutionBackwardWeightKernel< GroupedConvTraitsType_, TilePartitioner_, GemmPipeline_, EpiloguePipeline_ >::SplitKBatchOffset::a_k_split_offset

◆ b_k_split_offset

template<typename GroupedConvTraitsType_, typename TilePartitioner_, typename GemmPipeline_, typename EpiloguePipeline_>
index_t ck_tile::GroupedConvolutionBackwardWeightKernel< GroupedConvTraitsType_, TilePartitioner_, GemmPipeline_, EpiloguePipeline_ >::SplitKBatchOffset::b_k_split_offset

◆ splitted_k

template<typename GroupedConvTraitsType_, typename TilePartitioner_, typename GemmPipeline_, typename EpiloguePipeline_>
index_t ck_tile::GroupedConvolutionBackwardWeightKernel< GroupedConvTraitsType_, TilePartitioner_, GemmPipeline_, EpiloguePipeline_ >::SplitKBatchOffset::splitted_k
The documentation for this struct was generated from the following file: