StreamKTilePartitioner_v2< BlockGemmShapeType, ReductionStrategyType, Persistent > Struct Template Reference#
ck_tile::StreamKTilePartitioner_v2< BlockGemmShapeType, ReductionStrategyType, Persistent > Struct Template Reference
Template for the Stream-K tile partitioner derived struct. More...
Detailed Description
template<typename BlockGemmShapeType, StreamKReductionStrategy ReductionStrategyType, bool Persistent>
struct ck_tile::StreamKTilePartitioner_v2< BlockGemmShapeType, ReductionStrategyType, Persistent >
struct ck_tile::StreamKTilePartitioner_v2< BlockGemmShapeType, ReductionStrategyType, Persistent >
Template for the Stream-K tile partitioner derived struct.
This partitioner is responsible for mapping workgroups to tiles in the C tensor for the Stream-K algorithm. This struct is derived from StreamKTilePartitionerBase<BlockGemmShapeType, ReductionStrategyType>. Behavior of the StreamKTilePartitioner based on persistency will be in the template specializations.
- Template Parameters
-
BlockGemmShapeType A class providing basic GEMM parameters. ReductionStrategyType An enum that defines the reduction strategy for the results in the C Tensor. Persistent A bool that indicates whether to use a Persistent approach
The documentation for this struct was generated from the following file: