bert::FusedMultiHeadAttentionKernelMetaInfoV1 Struct Reference

Collaboration diagram for bert::FusedMultiHeadAttentionKernelMetaInfoV1:

Public Attributes
Data_type	mDataType
unsigned int	mS
unsigned int	mD
unsigned int	mSM
const unsigned char *	mCubin
unsigned int	mCubinSize
const char *	mFuncName
unsigned int	mSharedMemBytes
unsigned int	mThreadsPerCTA

Member Data Documentation

mDataType

Data_type bert::FusedMultiHeadAttentionKernelMetaInfoV1::mDataType

mS

unsigned int bert::FusedMultiHeadAttentionKernelMetaInfoV1::mS

mD

unsigned int bert::FusedMultiHeadAttentionKernelMetaInfoV1::mD

mSM

unsigned int bert::FusedMultiHeadAttentionKernelMetaInfoV1::mSM

mCubin

const unsigned char* bert::FusedMultiHeadAttentionKernelMetaInfoV1::mCubin

mCubinSize

unsigned int bert::FusedMultiHeadAttentionKernelMetaInfoV1::mCubinSize

mFuncName

const char* bert::FusedMultiHeadAttentionKernelMetaInfoV1::mFuncName

mSharedMemBytes

unsigned int bert::FusedMultiHeadAttentionKernelMetaInfoV1::mSharedMemBytes

mThreadsPerCTA

unsigned int bert::FusedMultiHeadAttentionKernelMetaInfoV1::mThreadsPerCTA

---

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

• fused_multihead_attention.h