mlir.dialects._x86vector_transform_ops_gen¶

Attributes¶

_ods_ir

Classes¶

`ApplyShuffleVectorFMAOpsPatternsOp`	Collect patterns to shuffle FMAs with x86vector operations as operands
`ApplyShuffleVectorFMAOpsPatternsOpAdaptor`
`ApplySinkVectorProducerOpsPatternsOp`	Collect patterns to sink vector producer operations forward in a block to
`ApplySinkVectorProducerOpsPatternsOpAdaptor`
`ApplyVectorContractBF16ToFMAPatternsOp`	Collect patterns to lower a BF16 type vector.contract operation
`ApplyVectorContractBF16ToFMAPatternsOpAdaptor`
`ApplyVectorContractToFMAPatternsOp`	Collect patterns to lower a F32 type vector.contract operation to a FMA.
`ApplyVectorContractToFMAPatternsOpAdaptor`
`ApplyVectorContractToPackedTypeDotProductPatternsOp`	Collect patterns to lower a BF16/Int8 type vector.contract operation
`ApplyVectorContractToPackedTypeDotProductPatternsOpAdaptor`

Functions¶

`apply_patterns_x86vector_shuffle_vector_fma_ops`(...)
`apply_patterns_x86vector_sink_vector_producer_ops`(...)
`apply_patterns_x86vector_vector_contract_bf16_to_fma`(...)
`apply_patterns_x86vector_vector_contract_to_fma`(...)
`apply_patterns_x86vector_vector_contract_to_packed_type_dot_product`(...)

Module Contents¶

mlir.dialects._x86vector_transform_ops_gen._ods_ir¶

class mlir.dialects._x86vector_transform_ops_gen.ApplyShuffleVectorFMAOpsPatternsOp(*, loc=None, ip=None)¶

Bases: _ods_ir

Collect patterns to shuffle FMAs with x86vector operations as operands such that FMAs are grouped with respect to odd/even packed index.

OPERATION_NAME = 'transform.apply_patterns.x86vector.shuffle_vector_fma_ops'¶

_ODS_REGIONS = (0, True)¶

class mlir.dialects._x86vector_transform_ops_gen.ApplyShuffleVectorFMAOpsPatternsOpAdaptor(operands: list, attributes: OpAttributeMap)¶

class mlir.dialects._x86vector_transform_ops_gen.ApplyShuffleVectorFMAOpsPatternsOpAdaptor(operands: list, opview: OpView)

Bases: _ods_ir

OPERATION_NAME = 'transform.apply_patterns.x86vector.shuffle_vector_fma_ops'¶

mlir.dialects._x86vector_transform_ops_gen.apply_patterns_x86vector_shuffle_vector_fma_ops(*, loc=None, ip=None) → ApplyShuffleVectorFMAOpsPatternsOp¶

class mlir.dialects._x86vector_transform_ops_gen.ApplySinkVectorProducerOpsPatternsOp(*, loc=None, ip=None)¶

Bases: _ods_ir

Collect patterns to sink vector producer operations forward in a block to place them immediately before their first use.

OPERATION_NAME = 'transform.apply_patterns.x86vector.sink_vector_producer_ops'¶

_ODS_REGIONS = (0, True)¶

class mlir.dialects._x86vector_transform_ops_gen.ApplySinkVectorProducerOpsPatternsOpAdaptor(operands: list, attributes: OpAttributeMap)¶

class mlir.dialects._x86vector_transform_ops_gen.ApplySinkVectorProducerOpsPatternsOpAdaptor(operands: list, opview: OpView)

Bases: _ods_ir

OPERATION_NAME = 'transform.apply_patterns.x86vector.sink_vector_producer_ops'¶

mlir.dialects._x86vector_transform_ops_gen.apply_patterns_x86vector_sink_vector_producer_ops(*, loc=None, ip=None) → ApplySinkVectorProducerOpsPatternsOp¶

class mlir.dialects._x86vector_transform_ops_gen.ApplyVectorContractBF16ToFMAPatternsOp(*, loc=None, ip=None)¶

Bases: _ods_ir

Collect patterns to lower a BF16 type vector.contract operation to a FMA via emulation lowering using BF16 packed operations.

OPERATION_NAME = 'transform.apply_patterns.x86vector.vector_contract_bf16_to_fma'¶

_ODS_REGIONS = (0, True)¶

class mlir.dialects._x86vector_transform_ops_gen.ApplyVectorContractBF16ToFMAPatternsOpAdaptor(operands: list, attributes: OpAttributeMap)¶

class mlir.dialects._x86vector_transform_ops_gen.ApplyVectorContractBF16ToFMAPatternsOpAdaptor(operands: list, opview: OpView)

Bases: _ods_ir

OPERATION_NAME = 'transform.apply_patterns.x86vector.vector_contract_bf16_to_fma'¶

mlir.dialects._x86vector_transform_ops_gen.apply_patterns_x86vector_vector_contract_bf16_to_fma(*, loc=None, ip=None) → ApplyVectorContractBF16ToFMAPatternsOp¶

class mlir.dialects._x86vector_transform_ops_gen.ApplyVectorContractToFMAPatternsOp(*, loc=None, ip=None)¶

Bases: _ods_ir

Collect patterns to lower a F32 type vector.contract operation to a FMA.

OPERATION_NAME = 'transform.apply_patterns.x86vector.vector_contract_to_fma'¶

_ODS_REGIONS = (0, True)¶

class mlir.dialects._x86vector_transform_ops_gen.ApplyVectorContractToFMAPatternsOpAdaptor(operands: list, attributes: OpAttributeMap)¶

class mlir.dialects._x86vector_transform_ops_gen.ApplyVectorContractToFMAPatternsOpAdaptor(operands: list, opview: OpView)

Bases: _ods_ir

OPERATION_NAME = 'transform.apply_patterns.x86vector.vector_contract_to_fma'¶

mlir.dialects._x86vector_transform_ops_gen.apply_patterns_x86vector_vector_contract_to_fma(*, loc=None, ip=None) → ApplyVectorContractToFMAPatternsOp¶

class mlir.dialects._x86vector_transform_ops_gen.ApplyVectorContractToPackedTypeDotProductPatternsOp(*, loc=None, ip=None)¶

Bases: _ods_ir

Collect patterns to lower a BF16/Int8 type vector.contract operation to a BF16/Int8 dot-product.

OPERATION_NAME = 'transform.apply_patterns.x86vector.vector_contract_to_packed_type_dot_product'¶

_ODS_REGIONS = (0, True)¶

class mlir.dialects._x86vector_transform_ops_gen.ApplyVectorContractToPackedTypeDotProductPatternsOpAdaptor(operands: list, attributes: OpAttributeMap)¶

class mlir.dialects._x86vector_transform_ops_gen.ApplyVectorContractToPackedTypeDotProductPatternsOpAdaptor(operands: list, opview: OpView)

Bases: _ods_ir

OPERATION_NAME = 'transform.apply_patterns.x86vector.vector_contract_to_packed_type_dot_product'¶

mlir.dialects._x86vector_transform_ops_gen.apply_patterns_x86vector_vector_contract_to_packed_type_dot_product(*, loc=None, ip=None) → ApplyVectorContractToPackedTypeDotProductPatternsOp¶