Back to Cutlass

CUTLASS: cutlass::layout::PackedVectorLayout Class Reference

docs/classcutlass_1_1layout_1_1PackedVectorLayout.html

4.4.27.6 KB
Original Source

| | CUTLASS

CUDA Templates for Linear Algebra Subroutines and Solvers |

Public Types | Public Member Functions | Static Public Member Functions | Static Public Attributes | List of all members

cutlass::layout::PackedVectorLayout Class Reference

Tensor layout for densely packed vectors.

#include <vector.h>

|

Public Types

| | using | Index = int32_t | | | Index type used for coordinates. More...
| | | | using | LongIndex = int64_t | | | Long index type used for offsets. More...
| | | | using | TensorCoord = Coord< kRank, Index > | | | Logical coordinate. More...
| | | | using | Stride = Coord< kStrideRank, Index > | | | Stride vector. More...
| | |

|

Public Member Functions

| | CUTLASS_HOST_DEVICE | PackedVectorLayout () | | | | CUTLASS_HOST_DEVICE LongIndex | operator() (TensorCoord const &coord) const | | | Returns the offset of a coordinate in linear memory. More...
| | | | CUTLASS_HOST_DEVICE Stride | stride () const | | | Returns the stride of the layout. More...
| | | | CUTLASS_HOST_DEVICE LongIndex | capacity (TensorCoord const &size) const | | | Compute the number of contiguous elements needed to store a tensor with the given size. More...
| | |

|

Static Public Member Functions

| | static CUTLASS_HOST_DEVICE PackedVectorLayout | packed (TensorCoord const &size) | | | Helper returns a layout to a tightly packed tensor. More...
| | |

|

Static Public Attributes

| | static int const | kRank = 1 | | | Logical rank of tensor. More...
| | | | static int const | kStrideRank = 1 | | | Rank of stride vector. More...
| | |

Member Typedef Documentation

| using cutlass::layout::PackedVectorLayout::Index = int32_t |

| using cutlass::layout::PackedVectorLayout::LongIndex = int64_t |

| using cutlass::layout::PackedVectorLayout::Stride = Coord<kStrideRank, Index> |

| using cutlass::layout::PackedVectorLayout::TensorCoord = Coord<kRank, Index> |

Constructor & Destructor Documentation

|

| CUTLASS_HOST_DEVICE cutlass::layout::PackedVectorLayout::PackedVectorLayout | ( | | ) | |

| inline |

Member Function Documentation

|

| CUTLASS_HOST_DEVICE LongIndex cutlass::layout::PackedVectorLayout::capacity | ( | TensorCoord const & | size | ) | const |

| inline |

|

| CUTLASS_HOST_DEVICE LongIndex cutlass::layout::PackedVectorLayout::operator() | ( | TensorCoord const & | coord | ) | const |

| inline |

|

| static CUTLASS_HOST_DEVICE PackedVectorLayout cutlass::layout::PackedVectorLayout::packed | ( | TensorCoord const & | size | ) | |

| inlinestatic |

|

| CUTLASS_HOST_DEVICE Stride cutlass::layout::PackedVectorLayout::stride | ( | | ) | const |

| inline |

Member Data Documentation

|

| int const cutlass::layout::PackedVectorLayout::kRank = 1 |

| static |

|

| int const cutlass::layout::PackedVectorLayout::kStrideRank = 1 |

| static |


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


Generated by 1.8.11