CUDA Templates for Linear Algebra Subroutines and Solvers
Public Attributes | List of all members
cutlass::library::GemmBatchedConfiguration Struct Reference

Configuration for batched GEMM in which multiple matrix products are computed.

#include <library.h>

Collaboration diagram for cutlass::library::GemmBatchedConfiguration:
Collaboration graph

Public Attributes

gemm::GemmCoord problem_size
 GEMM problem size. More...
int64_t lda
 Leading dimension of A matrix. More...
int64_t ldb
 Leading dimension of B matrix. More...
int64_t ldc
 Leading dimension of C matrix. More...
int64_t ldd
 Leading dimension of D matrix. More...
int64_t batch_stride_A
 Stride between instances of the A matrix in memory. More...
int64_t batch_stride_B
 Stride between instances of the B matrix in memory. More...
int64_t batch_stride_C
 Stride between instances of the C matrix in memory. More...
int64_t batch_stride_D
 Stride between instances of the D matrix in memory. More...
int batch_count
 Number of GEMMs in batch. More...

Member Data Documentation

int cutlass::library::GemmBatchedConfiguration::batch_count
int64_t cutlass::library::GemmBatchedConfiguration::batch_stride_A
int64_t cutlass::library::GemmBatchedConfiguration::batch_stride_B
int64_t cutlass::library::GemmBatchedConfiguration::batch_stride_C
int64_t cutlass::library::GemmBatchedConfiguration::batch_stride_D
int64_t cutlass::library::GemmBatchedConfiguration::lda
int64_t cutlass::library::GemmBatchedConfiguration::ldb
int64_t cutlass::library::GemmBatchedConfiguration::ldc
int64_t cutlass::library::GemmBatchedConfiguration::ldd
gemm::GemmCoord cutlass::library::GemmBatchedConfiguration::problem_size

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