kernel/gemm_splitk_parallel.h File Reference

Template for GEMM performing a reduction over K partitions in parallel. More...

#include "cutlass/cutlass.h"
#include "cutlass/gemm/gemm.h"
#include "cutlass/matrix_coord.h"

Include dependency graph for kernel/gemm_splitk_parallel.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes
struct	cutlass::gemm::kernel::GemmSplitKParallel< Mma_, Epilogue_, ThreadblockSwizzle_ >
struct	cutlass::gemm::kernel::GemmSplitKParallel< Mma_, Epilogue_, ThreadblockSwizzle_ >::Params
	Parameters structure. More...
union	cutlass::gemm::kernel::GemmSplitKParallel< Mma_, Epilogue_, ThreadblockSwizzle_ >::SharedStorage
	Shared memory storage structure. More...

Namespaces
	cutlass
	cutlass::gemm
	cutlass::gemm::kernel