XQA Kernel Types

struct XQAKernelMetaInfo

Public Types

enum XQAKernelVariant

Values:

enumerator KERNEL_VARIANT_STANDARD

enumerator KERNEL_VARIANT_FULL_SMEM_HEAD_DIM512

enumerator KERNEL_VARIANT_FULL_SMEM_HEAD_DIM512_ROW_MAX_METHOD4

enumerator KERNEL_VARIANT_TILED_QKV_STAGING_HEAD_DIM512

enumerator KERNEL_VARIANT_2CTA_HEAD_DIM512