mlir.dialects.linalg.opdsl.lang.dsl¶

Attributes¶

`_CONTEXT`
`StructuredOpOuts`

Classes¶

DefinedOpCallable

Callable that wraps any defined op function.

Functions¶

`bind_op_def`(op_def)
`current_op_def`(...)
`_prepare_structured_op_outs`(...)
`linalg_structured_op`(→ DefinedOpCallable)
`domain`(*dimensions)
`implements`(*interfaces)
`defines`(*definitions)

Module Contents¶

mlir.dialects.linalg.opdsl.lang.dsl._CONTEXT¶

mlir.dialects.linalg.opdsl.lang.dsl.StructuredOpOuts¶

mlir.dialects.linalg.opdsl.lang.dsl.bind_op_def(op_def: mlir.dialects.linalg.opdsl.lang.emitter.LinalgOpDef)¶

mlir.dialects.linalg.opdsl.lang.dsl.current_op_def() → mlir.dialects.linalg.opdsl.lang.emitter.LinalgOpDef¶

mlir.dialects.linalg.opdsl.lang.dsl._prepare_structured_op_outs(outs: StructuredOpOuts) → mlir.dialects.linalg.opdsl.lang.emitter.ValueList¶

class mlir.dialects.linalg.opdsl.lang.dsl.DefinedOpCallable(op_name: str, op_def: mlir.dialects.linalg.opdsl.lang.emitter.LinalgOpDef)¶

Callable that wraps any defined op function.

op_name¶

op_def¶

__call__(*ins: mlir.dialects.linalg.opdsl.lang.emitter.Union[mlir.ir.Operation, mlir.ir.OpView, mlir.ir.Value], outs: StructuredOpOuts, **kwargs)¶

Emits the corresponding op definition as IR.

Most arguments are passed through to the underlying emitter. The following keyword argument is interpreted here: emit_generic: Emits a generic form as appropriate (default True). If False, a named form is emitted (which must have been built in to the compiler).

mlir.dialects.linalg.opdsl.lang.dsl.linalg_structured_op(dsl_func=None, *, op_name=None, op_class_name=None) → DefinedOpCallable¶

mlir.dialects.linalg.opdsl.lang.dsl.domain(*dimensions: mlir.dialects.linalg.opdsl.lang.emitter.DimDef)¶

mlir.dialects.linalg.opdsl.lang.dsl.implements(*interfaces: mlir.dialects.linalg.opdsl.lang.emitter.OpInterfaceDef)¶

mlir.dialects.linalg.opdsl.lang.dsl.defines(*definitions: mlir.dialects.linalg.opdsl.lang.emitter.OpDefinitionDef)¶