CUTLASS: gemm_splitk_parallel.h File Reference

| | CUTLASS

CUDA Templates for Linear Algebra Subroutines and Solvers |

• Main Page

• Modules

• Namespaces

• Classes

• Files

• File List

• File Members

• include

• cutlass

• gemm

• kernel

Classes | Namespaces

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.](kernel_2gemm splitk parallel_8h_source.html)

|

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 | | |

---

Generated by 1.8.11