docs/structcutlass_1_1epilogue_1_1EpilogueWorkspace_1_1Params.html
| | CUTLASS
CUDA Templates for Linear Algebra Subroutines and Solvers |
Public Member Functions | Public Attributes | List of all members
cutlass::epilogue::EpilogueWorkspace< Shape_, WarpCount, FragmentC_ >::Params Struct Reference
Parameters structure.
#include <epilogue_workspace.h>
|
| | CUTLASS_HOST_DEVICE | Params (ElementC *ptr_C, int stride_n_, int stride_k_) | | |
|
|
| ElementC * | ptr_C |
| | Pointer to C matrix. More...
|
| |
| int | stride_n |
| | Stride between tiles along the GEMM N dimension (in units of vectors) More...
|
| |
| int | stride_k |
| | Stride between tiles along the GEMM K dimension (in units of vectors) More...
|
| |
template<typename Shape_ , int WarpCount, typename FragmentC_ >
|
| CUTLASS_HOST_DEVICE cutlass::epilogue::EpilogueWorkspace< Shape_, WarpCount, FragmentC_ >::Params::Params | ( | ElementC * | ptr_C, | | | | int | stride_n_, | | | | int | stride_k_ | | | ) | | |
| inline |
Parameters
| ptr_C | Pointer to C matrix | | stride_n_ | Stride between tiles along the GEMM N dimension (in units of ElementC) | | stride_k_ | Stride between tiles along the GEMM K dimension (in units of ElementC) |
template<typename Shape_ , int WarpCount, typename FragmentC_ >
| ElementC* cutlass::epilogue::EpilogueWorkspace< Shape_, WarpCount, FragmentC_ >::Params::ptr_C |
template<typename Shape_ , int WarpCount, typename FragmentC_ >
| int cutlass::epilogue::EpilogueWorkspace< Shape_, WarpCount, FragmentC_ >::Params::stride_k |
template<typename Shape_ , int WarpCount, typename FragmentC_ >
| int cutlass::epilogue::EpilogueWorkspace< Shape_, WarpCount, FragmentC_ >::Params::stride_n |
The documentation for this struct was generated from the following file:
Generated by 1.8.11