docs/structcutlass_1_1gemm_1_1BatchedGemmCoord.html
| | CUTLASS
CUDA Templates for Linear Algebra Subroutines and Solvers |
Public Types | Public Member Functions | Static Public Attributes | List of all members
cutlass::gemm::BatchedGemmCoord Struct Reference
#include <gemm.h>
Inheritance diagram for cutlass::gemm::BatchedGemmCoord:
 [legend]
Collaboration diagram for cutlass::gemm::BatchedGemmCoord:
 [legend]
|
|
| typedef int | Index |
| | Integer-valued index. More...
|
| |
| typedef Coord< 4, Index > | Base |
| | Base type is a Coord of rank=4. More...
|
| |
| Public Types inherited from cutlass::Coord< 4, int > |
| using | Index = int |
| | Index type used to store elements. More...
|
| |
| using | LongIndex = int64_t |
| | Type used to represent linear offsets. More...
|
| |
|
|
| CUTLASS_HOST_DEVICE | BatchedGemmCoord () |
| | Default ctor. More...
|
| |
| CUTLASS_HOST_DEVICE | BatchedGemmCoord (Base const &coord) |
| | Constructs from Coord<4>More...
|
| |
| CUTLASS_HOST_DEVICE | BatchedGemmCoord (Index m, Index n, Index k, Index b) |
| | Helper to construct from a K, N, M, and batch variables. More...
|
| |
| CUTLASS_HOST_DEVICE Index const & | m () const |
| | Returns the GEMM M coordinate. More...
|
| |
| CUTLASS_HOST_DEVICE Index & | m () |
| | Returns reference to the GEMM M coordinate. More...
|
| |
| CUTLASS_HOST_DEVICE Index const & | n () const |
| | Returns the GEMM N coordinate. More...
|
| |
| CUTLASS_HOST_DEVICE Index & | n () |
| | Returns reference to the GEMM N coordinate. More...
|
| |
| CUTLASS_HOST_DEVICE Index const & | k () const |
| | Returns the GEMM K coordinate. More...
|
| |
| CUTLASS_HOST_DEVICE Index & | k () |
| | Returns reference to the GEMM K coordinate. More...
|
| |
| CUTLASS_HOST_DEVICE Index const & | batch () const |
| | Returns the GEMM batch coordinate. More...
|
| |
| CUTLASS_HOST_DEVICE Index & | batch () |
| | Returns reference to the GEMM batch coordinate. More...
|
| |
| CUTLASS_HOST_DEVICE GemmCoord | mnk () const |
| | Obtains a GemmCoord from BatchedGemmCoord. More...
|
| |
| CUTLASS_HOST_DEVICE Coord< 4 > | mnkb () const |
| | Obtains a Coord<4> from BatchedGemmCoord. More...
|
| |
| CUTLASS_HOST_DEVICE BatchedGemmCoord | operator+ (Base const &b) const |
| | Element-wise addition. More...
|
| |
| CUTLASS_HOST_DEVICE BatchedGemmCoord | operator- (Base const &b) const |
| | Element-wise subtraction. More...
|
| |
| CUTLASS_HOST_DEVICE BatchedGemmCoord | operator* (Base const &b) const |
| | Element-wise multiplication. More...
|
| |
| CUTLASS_HOST_DEVICE BatchedGemmCoord | operator/ (Base const &b) const |
| | Element-wise division. More...
|
| |
| CUTLASS_HOST_DEVICE BatchedGemmCoord & | operator+= (Base const &b) |
| | In-place addition. More...
|
| |
| CUTLASS_HOST_DEVICE BatchedGemmCoord & | operator-= (Base const &b) |
| | In-place subtraction. More...
|
| |
| CUTLASS_HOST_DEVICE BatchedGemmCoord & | operator*= (Base const &b) |
| | In-place multiplication. More...
|
| |
| CUTLASS_HOST_DEVICE BatchedGemmCoord & | operator/= (Base const &b) |
| | In-place division. More...
|
| |
| Public Member Functions inherited from cutlass::Coord< 4, int > |
| CUTLASS_HOST_DEVICE | Coord (Index value=Index(0)) |
| | Default ctor initializes uniformly. More...
|
| |
| CUTLASS_HOST_DEVICE | Coord (Index const (&_idx)[kRank]) |
| | Constructs from an array of integers. More...
|
| |
| CUTLASS_HOST_DEVICE | Coord (Coord< kRank, Index, LongIndex > const &coord) |
| | Copy constructor. More...
|
| |
| CUTLASS_HOST_DEVICE Coord< Slice > | slice (int start=0, Index identity=0) const |
| |
| CUTLASS_HOST_DEVICE int | min_dim_index () const |
| | Returns the index of the dimension with least value. More...
|
| |
| CUTLASS_HOST_DEVICE int | max_dim_index () const |
| | Returns the index of the dimension with greatest value. More...
|
| |
| CUTLASS_HOST_DEVICE | operator bool () const |
| | Returns true if Coord is non-zero. More...
|
| |
| CUTLASS_HOST_DEVICE bool | operator! () const |
| | Returns true if Coord is uniformly zero. More...
|
| |
| CUTLASS_HOST_DEVICE Coord | operator+ (Coord const &b) const |
| | Element-wise addition. More...
|
| |
| CUTLASS_HOST_DEVICE Coord | operator- (Coord const &b) const |
| | Element-wise subtraction. More...
|
| |
| CUTLASS_HOST_DEVICE Coord | operator* (Coord const &b) const |
| | Element-wise multiplication. More...
|
| |
| CUTLASS_HOST_DEVICE Coord | operator/ (Coord const &b) const |
| | Element-wise division. More...
|
| |
| CUTLASS_HOST_DEVICE Coord & | operator+= (Coord const &b) |
| | In-place addition. More...
|
| |
| CUTLASS_HOST_DEVICE Coord & | operator-= (Coord const &b) |
| | In-place subtraction. More...
|
| |
| CUTLASS_HOST_DEVICE Coord & | operator*= (Coord const &b) |
| | In-place multiplication. More...
|
| |
| CUTLASS_HOST_DEVICE Coord & | operator/= (Coord const &b) |
| | In-place division. More...
|
| |
| CUTLASS_HOST_DEVICE Index & | operator[] (int dim) |
| | Member access operator. More...
|
| |
| CUTLASS_HOST_DEVICE Index const & | operator[] (int dim) const |
| | Member access operator. More...
|
| |
| CUTLASS_HOST_DEVICE LongIndex | dot (Coord const &b, LongIndex sum=LongIndex(0)) const |
| | Computes the dot product with anotherCoord object. More...
|
| |
| CUTLASS_HOST_DEVICE Index & | at () |
| | Gets the index of a given Coord element. More...
|
| |
| CUTLASS_HOST_DEVICE Index & | at (int dim) |
| | Access via index; may limit unrolling potential. More...
|
| |
| CUTLASS_HOST_DEVICE Index const & | at () const |
| | Gets the index of a given Coord element. More...
|
| |
| CUTLASS_HOST_DEVICE Index const & | at (int dim) const |
| | Access via index; may limit unrolling potential. More...
|
| |
| CUTLASS_HOST_DEVICE bool | operator== (Coord const &b) const |
| | Determines if two Coord<> objects are equal. More...
|
| |
| CUTLASS_HOST_DEVICE bool | operator!= (Coord const &b) const |
| | Not equal. More...
|
| |
| CUTLASS_HOST_DEVICE Coord & | clamp (Coord const &max, Coord const &min=Coord()) |
| | Clamps a coordinate to a range specified by maximum and minimum values. More...
|
| |
| CUTLASS_HOST_DEVICE Index | sum () const |
| | Returns the sum of all elements. More...
|
| |
| CUTLASS_HOST_DEVICE LongIndex | product () const |
| | Returns the product of all elements. More...
|
| |
| CUTLASS_HOST_DEVICE bool | operator< (Coord const &b) const |
| | Less than operator. More...
|
| |
| CUTLASS_HOST_DEVICE bool | operator<= (Coord const &b) const |
| | Less than or equals operator. More...
|
| |
| CUTLASS_HOST_DEVICE bool | operator> (Coord const &b) const |
| | Greater than operator. More...
|
| |
| CUTLASS_HOST_DEVICE bool | operator>= (Coord const &b) const |
| | Greater than or equals operator. More...
|
| |
|
|
| static int const | kM = 0 |
| | GEMM M dimension - rows of the output C matrix. More...
|
| |
| static int const | kN = 1 |
| | GEMM N dimension - columns of the output C matrix. More...
|
| |
| static int const | kK = 2 |
| | GEMM K dimension - inner dimension of the GEMM problem. More...
|
| |
| static int const | kBatch = 3 |
| | GEMM K dimension - inner dimension of the GEMM problem. More...
|
| |
| Static Public Attributes inherited from cutlass::Coord< 4, int > |
| static int const | kRank |
| | Number of elements in Coord. More...
|
| |
BatchedGemmCoord is a structure derived from Coord<4> that specifies a location within the coordinate space of a batched GEMM problem.
| typedef Coord<4, Index> cutlass::gemm::BatchedGemmCoord::Base |
| typedef int cutlass::gemm::BatchedGemmCoord::Index |
|
| CUTLASS_HOST_DEVICE cutlass::gemm::BatchedGemmCoord::BatchedGemmCoord | ( | | ) | |
| inline |
|
| CUTLASS_HOST_DEVICE cutlass::gemm::BatchedGemmCoord::BatchedGemmCoord | ( | Base const & | coord | ) | |
| inline |
|
| CUTLASS_HOST_DEVICE cutlass::gemm::BatchedGemmCoord::BatchedGemmCoord | ( | Index | m, | | | | Index | n, | | | | Index | k, | | | | Index | b | | | ) | | |
| inline |
|
| CUTLASS_HOST_DEVICE Index const& cutlass::gemm::BatchedGemmCoord::batch | ( | | ) | const |
| inline |
|
| CUTLASS_HOST_DEVICE Index& cutlass::gemm::BatchedGemmCoord::batch | ( | | ) | |
| inline |
|
| CUTLASS_HOST_DEVICE Index const& cutlass::gemm::BatchedGemmCoord::k | ( | | ) | const |
| inline |
|
| CUTLASS_HOST_DEVICE Index& cutlass::gemm::BatchedGemmCoord::k | ( | | ) | |
| inline |
|
| CUTLASS_HOST_DEVICE Index const& cutlass::gemm::BatchedGemmCoord::m | ( | | ) | const |
| inline |
|
| CUTLASS_HOST_DEVICE Index& cutlass::gemm::BatchedGemmCoord::m | ( | | ) | |
| inline |
|
| CUTLASS_HOST_DEVICE GemmCoord cutlass::gemm::BatchedGemmCoord::mnk | ( | | ) | const |
| inline |
|
| CUTLASS_HOST_DEVICE Coord<4> cutlass::gemm::BatchedGemmCoord::mnkb | ( | | ) | const |
| inline |
|
| CUTLASS_HOST_DEVICE Index const& cutlass::gemm::BatchedGemmCoord::n | ( | | ) | const |
| inline |
|
| CUTLASS_HOST_DEVICE Index& cutlass::gemm::BatchedGemmCoord::n | ( | | ) | |
| inline |
|
| CUTLASS_HOST_DEVICE BatchedGemmCoord cutlass::gemm::BatchedGemmCoord::operator* | ( | Base const & | b | ) | const |
| inline |
|
| CUTLASS_HOST_DEVICE BatchedGemmCoord& cutlass::gemm::BatchedGemmCoord::operator*= | ( | Base const & | b | ) | |
| inline |
|
| CUTLASS_HOST_DEVICE BatchedGemmCoord cutlass::gemm::BatchedGemmCoord::operator+ | ( | Base const & | b | ) | const |
| inline |
|
| CUTLASS_HOST_DEVICE BatchedGemmCoord& cutlass::gemm::BatchedGemmCoord::operator+= | ( | Base const & | b | ) | |
| inline |
|
| CUTLASS_HOST_DEVICE BatchedGemmCoord cutlass::gemm::BatchedGemmCoord::operator- | ( | Base const & | b | ) | const |
| inline |
|
| CUTLASS_HOST_DEVICE BatchedGemmCoord& cutlass::gemm::BatchedGemmCoord::operator-= | ( | Base const & | b | ) | |
| inline |
|
| CUTLASS_HOST_DEVICE BatchedGemmCoord cutlass::gemm::BatchedGemmCoord::operator/ | ( | Base const & | b | ) | const |
| inline |
|
| CUTLASS_HOST_DEVICE BatchedGemmCoord& cutlass::gemm::BatchedGemmCoord::operator/= | ( | Base const & | b | ) | |
| inline |
|
| int const cutlass::gemm::BatchedGemmCoord::kBatch = 3 |
| static |
|
| int const cutlass::gemm::BatchedGemmCoord::kK = 2 |
| static |
|
| int const cutlass::gemm::BatchedGemmCoord::kM = 0 |
| static |
|
| int const cutlass::gemm::BatchedGemmCoord::kN = 1 |
| static |
The documentation for this struct was generated from the following file:
Generated by 1.8.11