|
Mila
Deep Neural Network Library
|
Files | |
| GqaState.ixx | |
| Non-owning transient scratch state for CudaGqaOp inference paths. | |
| IKVCacheLifecycle.ixx | |
| Interface for operations that own and manage a KV cache. | |
| IKvInference.ixx | |
| KV-cache compute interface for modern attention backends (GQA and beyond). | |
| IPackedKvInference.ixx | |
| KV-cache inference interface for GPT-style packed-QKV MHA backends. | |
| IPositionalDecode.ixx | |
| Interface for unary operations whose output depends on absolute token position. | |
| IPositionalPairedOp.ixx | |
| Interface for paired operations whose output depends on absolute token position. | |
| OperationBase.ixx | |
| Core abstraction for neural network operations in the Mila framework. | |
| OperationTraits.ixx | |
| Aggregator for the unified operation traits dispatch table. | |
| OperationTraits.Template.ixx | |
| Unified compile-time dispatch template mapping (OperationType, DeviceType, TPrecision, TPolicy) to a concrete operation type. | |
| OperationType.ixx | |
| Defines the operation types supported by the compute framework. | |