Threadblock swizzling function for split-K GEMMs.

Public Member Functions
CUTLASS_HOST_DEVICE GemmCoord	get_tiled_shape (GemmCoord problem_size, GemmCoord tile_size, int partitions) const
	Returns the shape of the problem in units of logical tiles. More...

CUTLASS_HOST_DEVICE dim3	get_grid_shape (GemmCoord tiled_shape) const
	Computes CUDA grid dimensions given a size in units of logical tiles. More...

CUTLASS_DEVICE GemmCoord	get_tile_offset () const
	Obtains the threadblock offset (in units of threadblock-scoped tiles) More...

Member Function Documentation

CUTLASS_HOST_DEVICE dim3 cutlass::gemm::threadblock::GemmSplitKHorizontalThreadblockSwizzle::get_grid_shape ( GemmCoord tiled_shape ) const

inline

CUTLASS_DEVICE GemmCoord cutlass::gemm::threadblock::GemmSplitKHorizontalThreadblockSwizzle::get_tile_offset ( ) const

inline

CUTLASS_HOST_DEVICE GemmCoord cutlass::gemm::threadblock::GemmSplitKHorizontalThreadblockSwizzle::get_tiled_shape	(	GemmCoord	problem_size,
		GemmCoord	tile_size,
		int	partitions
	)		const

inline

The documentation for this struct was generated from the following file:

Public Member Functions