mlir.dialects.transform.xegpu

Classes

GetDescOp	Specialization for GetDescOp class.
SetDescLayoutOp	Specialization for SetDescLayoutOp class.
SetOpLayoutAttrOp	Specialization for SetOpLayoutAttrOp class.
SetGPULaunchThreadsOp	Specialization for SetGPULaunchThreadsOp class.
InsertPrefetchOp	Specialization for InsertPrefetchOp class.
ConvertLayoutOp	Specialization for ConvertLayoutOp class.

Functions

get_desc_op(→ OpResult)
set_desc_layout(→ OpResult)
set_op_layout_attr(→ SetOpLayoutAttrOp)
set_gpu_launch_threads(→ SetGPULaunchThreadsOp)
insert_prefetch(→ OpResult)
convert_layout(→ ConvertLayoutOp)

Module Contents

class mlir.dialects.transform.xegpu.GetDescOp(target: Value, *, loc=None, ip=None)

Bases: GetDescOp

Specialization for GetDescOp class.

mlir.dialects.transform.xegpu.get_desc_op(target: Value, *, loc=None, ip=None) → OpResult

class mlir.dialects.transform.xegpu.SetDescLayoutOp(target: Operation | Value, sg_layout: mlir.dialects._ods_common.MixedValues, sg_data: mlir.dialects._ods_common.MixedValues, *, inst_data: mlir.dialects._ods_common.MixedValues | None = None, loc=None, ip=None)

Bases: SetDescLayoutOp

Specialization for SetDescLayoutOp class.

mlir.dialects.transform.xegpu.set_desc_layout(target: Operation | Value, sg_layout: mlir.dialects._ods_common.MixedValues, sg_data: mlir.dialects._ods_common.MixedValues, *, inst_data: mlir.dialects._ods_common.MixedValues | None = None, loc=None, ip=None) → OpResult

class mlir.dialects.transform.xegpu.SetOpLayoutAttrOp(target: Operation | Value, sg_layout: mlir.dialects._ods_common.MixedValues, sg_data: mlir.dialects._ods_common.MixedValues, *, inst_data: mlir.dialects._ods_common.MixedValues | None = None, index: int | Attribute | None = None, result: bool | Attribute | None = None, loc=None, ip=None)

Bases: SetOpLayoutAttrOp

Specialization for SetOpLayoutAttrOp class.

mlir.dialects.transform.xegpu.set_op_layout_attr(target: Operation | Value, sg_layout: mlir.dialects._ods_common.MixedValues, sg_data: mlir.dialects._ods_common.MixedValues, *, inst_data: mlir.dialects._ods_common.MixedValues | None = None, index: int | Attribute | None = None, result: bool | Attribute | None = None, loc=None, ip=None) → SetOpLayoutAttrOp

class mlir.dialects.transform.xegpu.SetGPULaunchThreadsOp(launch_op: Operation | Value, threads: mlir.dialects._ods_common.MixedValues, *, loc=None, ip=None)

Bases: SetGPULaunchThreadsOp

Specialization for SetGPULaunchThreadsOp class.

mlir.dialects.transform.xegpu.set_gpu_launch_threads(launch_op: Operation | Value, threads: mlir.dialects._ods_common.MixedValues, *, loc=None, ip=None) → SetGPULaunchThreadsOp

class mlir.dialects.transform.xegpu.InsertPrefetchOp(target: Value, *, nb_prefetch: mlir.dialects._ods_common.MixedInt | None = 1, loc=None, ip=None)

Bases: InsertPrefetchOp

Specialization for InsertPrefetchOp class.

mlir.dialects.transform.xegpu.insert_prefetch(target: Value, *, nb_prefetch: mlir.dialects._ods_common.MixedInt | None = 1, loc=None, ip=None) → OpResult

class mlir.dialects.transform.xegpu.ConvertLayoutOp(target: Value, input_sg_layout: mlir.dialects._ods_common.MixedValues, input_sg_data: mlir.dialects._ods_common.MixedValues, target_sg_layout: mlir.dialects._ods_common.MixedValues, target_sg_data: mlir.dialects._ods_common.MixedValues, *, input_inst_data: mlir.dialects._ods_common.MixedValues | None = None, target_inst_data: mlir.dialects._ods_common.MixedValues | None = None, loc=None, ip=None)

Bases: ConvertLayoutOp

Specialization for ConvertLayoutOp class.

mlir.dialects.transform.xegpu.convert_layout(target: Value, input_sg_layout: mlir.dialects._ods_common.MixedValues, input_sg_data: mlir.dialects._ods_common.MixedValues, target_sg_layout: mlir.dialects._ods_common.MixedValues, target_sg_data: mlir.dialects._ods_common.MixedValues, *, input_inst_data: mlir.dialects._ods_common.MixedValues | None = None, target_inst_data: mlir.dialects._ods_common.MixedValues | None = None, loc=None, ip=None) → ConvertLayoutOp