Skip to main content Module features Copy item path Source CubeMmaConfig Shape and element types of a valid flexible MMA configuration
Only Vulkan for now, but this should also be usable for wgmma/xmma on datacenter CUDA.
Actual matrix size must be multiple of granularity and <= max. EnumSet A set of enum variants, stored as a bitmask. Features Features supported by a runtime MatmulFeatures Matrix multiplication-related features MmaConfig Shape and element types of a valid MMA configuration ScaledMmaConfig Shape and element types of a valid block-scaled MMA configuration Types Type support for a device AtomicUsage Atomic operations allowed for this type. ComplexUsage Complex capability families allowed for a complex element type. Plane Supported plane features Tma Atomic features that may be supported by a Runtime. TypeUsage Operations allowed for this type. CMMA is defined separately.