core.extensions.kitchen

Module Contents

Data

HAVE_KITCHEN
AutogradFunctionImplementation
KitchenSpecProvider
QAttentionParamsConfigSchema
QFlashAttentionParamsConfigSchema
QLinearParamsConfigSchema
QLinearParams
QuantizeRecipe
QuantizeRecipeAttnBMM
get_qattention_params_from_predefined
get_qfa_params_from_recipe_name
get_qlinear_params_from_predefined
get_qlinear_params_from_qat_params
KitchenColumnParallelGroupedLinear
KitchenColumnParallelLinear
KitchenDotProductAttention
KitchenFlashAttention
KitchenLayerNormColumnParallelLinear
KitchenRowParallelGroupedLinear
KitchenRowParallelLinear

API

core.extensions.kitchen.HAVE_KITCHEN

False

core.extensions.kitchen.AutogradFunctionImplementation

‘MagicMock(…)’

core.extensions.kitchen.KitchenSpecProvider

‘MagicMock(…)’

core.extensions.kitchen.QAttentionParamsConfigSchema

‘MagicMock(…)’

core.extensions.kitchen.QFlashAttentionParamsConfigSchema

‘MagicMock(…)’

core.extensions.kitchen.QLinearParamsConfigSchema

‘MagicMock(…)’

core.extensions.kitchen.QLinearParams

‘MagicMock(…)’

core.extensions.kitchen.QuantizeRecipe

‘MagicMock(…)’

core.extensions.kitchen.QuantizeRecipeAttnBMM

‘MagicMock(…)’

core.extensions.kitchen.get_qattention_params_from_predefined

‘MagicMock(…)’

core.extensions.kitchen.get_qfa_params_from_recipe_name

‘MagicMock(…)’

core.extensions.kitchen.get_qlinear_params_from_predefined

‘MagicMock(…)’

core.extensions.kitchen.get_qlinear_params_from_qat_params

‘MagicMock(…)’

core.extensions.kitchen.KitchenColumnParallelGroupedLinear

‘MagicMock(…)’

core.extensions.kitchen.KitchenColumnParallelLinear

‘MagicMock(…)’

core.extensions.kitchen.KitchenDotProductAttention

‘MagicMock(…)’

core.extensions.kitchen.KitchenFlashAttention

‘MagicMock(…)’

core.extensions.kitchen.KitchenLayerNormColumnParallelLinear

‘MagicMock(…)’

core.extensions.kitchen.KitchenRowParallelGroupedLinear

‘MagicMock(…)’

core.extensions.kitchen.KitchenRowParallelLinear

‘MagicMock(…)’