mlir.dialects.transform.x86vector ================================= .. py:module:: mlir.dialects.transform.x86vector Classes ------- .. autoapisummary:: mlir.dialects.transform.x86vector.ApplyShuffleVectorFMAOpsPatternsOp mlir.dialects.transform.x86vector.ApplyShuffleVectorFMAOpsPatternsOpAdaptor mlir.dialects.transform.x86vector.ApplySinkVectorProducerOpsPatternsOp mlir.dialects.transform.x86vector.ApplySinkVectorProducerOpsPatternsOpAdaptor mlir.dialects.transform.x86vector.ApplyVectorContractBF16ToFMAPatternsOp mlir.dialects.transform.x86vector.ApplyVectorContractBF16ToFMAPatternsOpAdaptor mlir.dialects.transform.x86vector.ApplyVectorContractToFMAPatternsOp mlir.dialects.transform.x86vector.ApplyVectorContractToFMAPatternsOpAdaptor mlir.dialects.transform.x86vector.ApplyVectorContractToPackedTypeDotProductPatternsOp mlir.dialects.transform.x86vector.ApplyVectorContractToPackedTypeDotProductPatternsOpAdaptor Functions --------- .. autoapisummary:: mlir.dialects.transform.x86vector.apply_patterns_x86vector_shuffle_vector_fma_ops mlir.dialects.transform.x86vector.apply_patterns_x86vector_sink_vector_producer_ops mlir.dialects.transform.x86vector.apply_patterns_x86vector_vector_contract_bf16_to_fma mlir.dialects.transform.x86vector.apply_patterns_x86vector_vector_contract_to_fma mlir.dialects.transform.x86vector.apply_patterns_x86vector_vector_contract_to_packed_type_dot_product Module Contents --------------- .. py:class:: ApplyShuffleVectorFMAOpsPatternsOp(*, loc=None, ip=None) Bases: :py:obj:`_ods_ir` Collect patterns to shuffle FMAs with x86vector operations as operands such that FMAs are grouped with respect to odd/even packed index. .. py:attribute:: OPERATION_NAME :value: 'transform.apply_patterns.x86vector.shuffle_vector_fma_ops' .. py:attribute:: _ODS_REGIONS :value: (0, True) .. py:class:: ApplyShuffleVectorFMAOpsPatternsOpAdaptor(operands: list, attributes: OpAttributeMap) ApplyShuffleVectorFMAOpsPatternsOpAdaptor(operands: list, opview: OpView) Bases: :py:obj:`_ods_ir` .. py:attribute:: OPERATION_NAME :value: 'transform.apply_patterns.x86vector.shuffle_vector_fma_ops' .. py:function:: apply_patterns_x86vector_shuffle_vector_fma_ops(*, loc=None, ip=None) -> ApplyShuffleVectorFMAOpsPatternsOp .. py:class:: ApplySinkVectorProducerOpsPatternsOp(*, loc=None, ip=None) Bases: :py:obj:`_ods_ir` Collect patterns to sink vector producer operations forward in a block to place them immediately before their first use. .. py:attribute:: OPERATION_NAME :value: 'transform.apply_patterns.x86vector.sink_vector_producer_ops' .. py:attribute:: _ODS_REGIONS :value: (0, True) .. py:class:: ApplySinkVectorProducerOpsPatternsOpAdaptor(operands: list, attributes: OpAttributeMap) ApplySinkVectorProducerOpsPatternsOpAdaptor(operands: list, opview: OpView) Bases: :py:obj:`_ods_ir` .. py:attribute:: OPERATION_NAME :value: 'transform.apply_patterns.x86vector.sink_vector_producer_ops' .. py:function:: apply_patterns_x86vector_sink_vector_producer_ops(*, loc=None, ip=None) -> ApplySinkVectorProducerOpsPatternsOp .. py:class:: ApplyVectorContractBF16ToFMAPatternsOp(*, loc=None, ip=None) Bases: :py:obj:`_ods_ir` Collect patterns to lower a BF16 type vector.contract operation to a FMA via emulation lowering using BF16 packed operations. .. py:attribute:: OPERATION_NAME :value: 'transform.apply_patterns.x86vector.vector_contract_bf16_to_fma' .. py:attribute:: _ODS_REGIONS :value: (0, True) .. py:class:: ApplyVectorContractBF16ToFMAPatternsOpAdaptor(operands: list, attributes: OpAttributeMap) ApplyVectorContractBF16ToFMAPatternsOpAdaptor(operands: list, opview: OpView) Bases: :py:obj:`_ods_ir` .. py:attribute:: OPERATION_NAME :value: 'transform.apply_patterns.x86vector.vector_contract_bf16_to_fma' .. py:function:: apply_patterns_x86vector_vector_contract_bf16_to_fma(*, loc=None, ip=None) -> ApplyVectorContractBF16ToFMAPatternsOp .. py:class:: ApplyVectorContractToFMAPatternsOp(*, loc=None, ip=None) Bases: :py:obj:`_ods_ir` Collect patterns to lower a F32 type vector.contract operation to a FMA. .. py:attribute:: OPERATION_NAME :value: 'transform.apply_patterns.x86vector.vector_contract_to_fma' .. py:attribute:: _ODS_REGIONS :value: (0, True) .. py:class:: ApplyVectorContractToFMAPatternsOpAdaptor(operands: list, attributes: OpAttributeMap) ApplyVectorContractToFMAPatternsOpAdaptor(operands: list, opview: OpView) Bases: :py:obj:`_ods_ir` .. py:attribute:: OPERATION_NAME :value: 'transform.apply_patterns.x86vector.vector_contract_to_fma' .. py:function:: apply_patterns_x86vector_vector_contract_to_fma(*, loc=None, ip=None) -> ApplyVectorContractToFMAPatternsOp .. py:class:: ApplyVectorContractToPackedTypeDotProductPatternsOp(*, loc=None, ip=None) Bases: :py:obj:`_ods_ir` Collect patterns to lower a BF16/Int8 type vector.contract operation to a BF16/Int8 dot-product. .. py:attribute:: OPERATION_NAME :value: 'transform.apply_patterns.x86vector.vector_contract_to_packed_type_dot_product' .. py:attribute:: _ODS_REGIONS :value: (0, True) .. py:class:: ApplyVectorContractToPackedTypeDotProductPatternsOpAdaptor(operands: list, attributes: OpAttributeMap) ApplyVectorContractToPackedTypeDotProductPatternsOpAdaptor(operands: list, opview: OpView) Bases: :py:obj:`_ods_ir` .. py:attribute:: OPERATION_NAME :value: 'transform.apply_patterns.x86vector.vector_contract_to_packed_type_dot_product' .. py:function:: apply_patterns_x86vector_vector_contract_to_packed_type_dot_product(*, loc=None, ip=None) -> ApplyVectorContractToPackedTypeDotProductPatternsOp