beaver

Beaver.MLIR.Transforms (beaver v0.2.19)

Transformations MLIR provides by default.

Link to this section Summary

Functions

canonicalize()

Canonicalize operations

canonicalize(composer_or_op)
control_flow_sink()

Sink operations into conditional blocks

control_flow_sink(composer_or_op)
convert_elementwise_to_linalg()

Convert ElementwiseMappable ops to linalg

convert_elementwise_to_linalg(composer_or_op)
convert_linalg_to_affine_loops()

Lower the operations from the linalg dialect into affine loops

convert_linalg_to_affine_loops(composer_or_op)
convert_linalg_to_loops()

Lower the operations from the linalg dialect into loops

convert_linalg_to_loops(composer_or_op)
convert_linalg_to_parallel_loops()

Lower the operations from the linalg dialect into parallel loops

convert_linalg_to_parallel_loops(composer_or_op)
cse()

Eliminate common sub-expressions

cse(composer_or_op)
generate_runtime_verification()

Generate additional runtime op verification checks

generate_runtime_verification(composer_or_op)
gpu_async_region()

Make GPU ops async

gpu_async_region(composer_or_op)
gpu_kernel_outlining()

Outline gpu.launch bodies to kernel functions

gpu_kernel_outlining(composer_or_op)
gpu_launch_sink_index_computations()

Sink index computations into gpu.launch body

gpu_launch_sink_index_computations(composer_or_op)
gpu_map_parallel_loops()

Greedily maps loops to GPU hardware dimensions.

gpu_map_parallel_loops(composer_or_op)
inline()

Inline function calls

inline(composer_or_op)
linalg_bufferize()

Bufferize the linalg dialect

linalg_bufferize(composer_or_op)
linalg_detensorize()

Detensorize linalg ops

linalg_detensorize(composer_or_op)
linalg_fold_unit_extent_dims()

Remove unit-extent dimension in Linalg ops on tensors

linalg_fold_unit_extent_dims(composer_or_op)
linalg_fuse_elementwise_ops()

Fuse elementwise operations on tensors

linalg_fuse_elementwise_ops(composer_or_op)
linalg_generalize_named_ops()

Convert named ops into generic ops

linalg_generalize_named_ops(composer_or_op)
linalg_inline_scalar_operands()

Inline scalar operands into linalg generic ops

linalg_inline_scalar_operands(composer_or_op)
linalg_named_op_conversion()

Convert from one named linalg op to another.

linalg_named_op_conversion(composer_or_op)
loop_invariant_code_motion()

Hoist loop invariant instructions outside of the loop

loop_invariant_code_motion(composer_or_op)
print_ir()

Print IR on the debug stream

print_ir(composer_or_op)
print_op_stats()

Print statistics of operations

print_op_stats(composer_or_op)
sccp()

Sparse Conditional Constant Propagation

sccp(composer_or_op)
snapshot_op_locations()

Generate new locations from the current IR

snapshot_op_locations(composer_or_op)
strip_debuginfo()

Strip debug info from all operations

strip_debuginfo(composer_or_op)
symbol_dce()

Eliminate dead symbols

symbol_dce(composer_or_op)
symbol_privatize()

Mark symbols private

symbol_privatize(composer_or_op)
topological_sort()

Sort regions without SSA dominance in topological order

topological_sort(composer_or_op)
view_op_graph()

Print Graphviz visualization of an operation

view_op_graph(composer_or_op)

Link to this section Functions

Link to this function

canonicalize()

Canonicalize operations

argument-name-in-mlir-cli

 Argument name in MLIR CLI

canonicalize

pass-name-in-tablegen

 Pass name in TableGen

Canonicalizer

Link to this function

canonicalize(composer_or_op)

Link to this function

control_flow_sink()

Sink operations into conditional blocks

argument-name-in-mlir-cli

 Argument name in MLIR CLI

control-flow-sink

pass-name-in-tablegen

 Pass name in TableGen

ControlFlowSink

Link to this function

control_flow_sink(composer_or_op)

Link to this function

convert_elementwise_to_linalg()

Convert ElementwiseMappable ops to linalg

argument-name-in-mlir-cli

 Argument name in MLIR CLI

convert-elementwise-to-linalg

pass-name-in-tablegen

 Pass name in TableGen

ConvertElementwiseToLinalg

Link to this function

convert_elementwise_to_linalg(composer_or_op)

Link to this function

convert_linalg_to_affine_loops()

Lower the operations from the linalg dialect into affine loops

argument-name-in-mlir-cli

 Argument name in MLIR CLI

convert-linalg-to-affine-loops

pass-name-in-tablegen

 Pass name in TableGen

LinalgLowerToAffineLoops

Link to this function

convert_linalg_to_affine_loops(composer_or_op)

Link to this function

convert_linalg_to_loops()

Lower the operations from the linalg dialect into loops

argument-name-in-mlir-cli

 Argument name in MLIR CLI

convert-linalg-to-loops

pass-name-in-tablegen

 Pass name in TableGen

LinalgLowerToLoops

Link to this function

convert_linalg_to_loops(composer_or_op)

Link to this function

convert_linalg_to_parallel_loops()

Lower the operations from the linalg dialect into parallel loops

argument-name-in-mlir-cli

 Argument name in MLIR CLI

convert-linalg-to-parallel-loops

pass-name-in-tablegen

 Pass name in TableGen

LinalgLowerToParallelLoops

Link to this function

convert_linalg_to_parallel_loops(composer_or_op)

Link to this function

cse()

Eliminate common sub-expressions

argument-name-in-mlir-cli

 Argument name in MLIR CLI

cse

pass-name-in-tablegen

 Pass name in TableGen

CSE

Link to this function

cse(composer_or_op)

Link to this function

generate_runtime_verification()

Generate additional runtime op verification checks

argument-name-in-mlir-cli

 Argument name in MLIR CLI

generate-runtime-verification

pass-name-in-tablegen

 Pass name in TableGen

GenerateRuntimeVerification

Link to this function

generate_runtime_verification(composer_or_op)

Link to this function

gpu_async_region()

Make GPU ops async

argument-name-in-mlir-cli

 Argument name in MLIR CLI

gpu-async-region

pass-name-in-tablegen

 Pass name in TableGen

GpuAsyncRegionPass

Link to this function

gpu_async_region(composer_or_op)

Link to this function

gpu_kernel_outlining()

Outline gpu.launch bodies to kernel functions

argument-name-in-mlir-cli

 Argument name in MLIR CLI

gpu-kernel-outlining

pass-name-in-tablegen

 Pass name in TableGen

GpuKernelOutlining

Link to this function

gpu_kernel_outlining(composer_or_op)

Link to this function

gpu_launch_sink_index_computations()

Sink index computations into gpu.launch body

argument-name-in-mlir-cli

 Argument name in MLIR CLI

gpu-launch-sink-index-computations

pass-name-in-tablegen

 Pass name in TableGen

GpuLaunchSinkIndexComputations

Link to this function

gpu_launch_sink_index_computations(composer_or_op)

Link to this function

gpu_map_parallel_loops()

Greedily maps loops to GPU hardware dimensions.

argument-name-in-mlir-cli

 Argument name in MLIR CLI

gpu-map-parallel-loops

pass-name-in-tablegen

 Pass name in TableGen

GpuMapParallelLoopsPass

Link to this function

gpu_map_parallel_loops(composer_or_op)

Link to this function

inline()

Inline function calls

argument-name-in-mlir-cli

 Argument name in MLIR CLI

inline

pass-name-in-tablegen

 Pass name in TableGen

Inliner

Link to this function

inline(composer_or_op)

Link to this function

linalg_bufferize()

Bufferize the linalg dialect

argument-name-in-mlir-cli

 Argument name in MLIR CLI

linalg-bufferize

pass-name-in-tablegen

 Pass name in TableGen

LinalgBufferize

Link to this function

linalg_bufferize(composer_or_op)

Link to this function

linalg_detensorize()

Detensorize linalg ops

argument-name-in-mlir-cli

 Argument name in MLIR CLI

linalg-detensorize

pass-name-in-tablegen

 Pass name in TableGen

LinalgDetensorize

Link to this function

linalg_detensorize(composer_or_op)

Link to this function

linalg_fold_unit_extent_dims()

Remove unit-extent dimension in Linalg ops on tensors

argument-name-in-mlir-cli

 Argument name in MLIR CLI

linalg-fold-unit-extent-dims

pass-name-in-tablegen

 Pass name in TableGen

LinalgFoldUnitExtentDims

Link to this function

linalg_fold_unit_extent_dims(composer_or_op)

Link to this function

linalg_fuse_elementwise_ops()

Fuse elementwise operations on tensors

argument-name-in-mlir-cli

 Argument name in MLIR CLI

linalg-fuse-elementwise-ops

pass-name-in-tablegen

 Pass name in TableGen

LinalgElementwiseOpFusion

Link to this function

linalg_fuse_elementwise_ops(composer_or_op)

Link to this function

linalg_generalize_named_ops()

Convert named ops into generic ops

argument-name-in-mlir-cli

 Argument name in MLIR CLI

linalg-generalize-named-ops

pass-name-in-tablegen

 Pass name in TableGen

LinalgGeneralization

Link to this function

linalg_generalize_named_ops(composer_or_op)

Link to this function

linalg_inline_scalar_operands()

Inline scalar operands into linalg generic ops

argument-name-in-mlir-cli

 Argument name in MLIR CLI

linalg-inline-scalar-operands

pass-name-in-tablegen

 Pass name in TableGen

LinalgInlineScalarOperands

Link to this function

linalg_inline_scalar_operands(composer_or_op)

Link to this function

linalg_named_op_conversion()

Convert from one named linalg op to another.

argument-name-in-mlir-cli

 Argument name in MLIR CLI

linalg-named-op-conversion

pass-name-in-tablegen

 Pass name in TableGen

LinalgNamedOpConversion

Link to this function

linalg_named_op_conversion(composer_or_op)

Link to this function

loop_invariant_code_motion()

Hoist loop invariant instructions outside of the loop

argument-name-in-mlir-cli

 Argument name in MLIR CLI

loop-invariant-code-motion

pass-name-in-tablegen

 Pass name in TableGen

LoopInvariantCodeMotion

Link to this function

loop_invariant_code_motion(composer_or_op)

Link to this function

sccp()

Sparse Conditional Constant Propagation

argument-name-in-mlir-cli

 Argument name in MLIR CLI

sccp

pass-name-in-tablegen

 Pass name in TableGen

SCCP

Link to this function

sccp(composer_or_op)

Link to this function

snapshot_op_locations()

Generate new locations from the current IR

argument-name-in-mlir-cli

 Argument name in MLIR CLI

snapshot-op-locations

pass-name-in-tablegen

 Pass name in TableGen

LocationSnapshot

Link to this function

snapshot_op_locations(composer_or_op)

Link to this function

strip_debuginfo()

Strip debug info from all operations

argument-name-in-mlir-cli

 Argument name in MLIR CLI

strip-debuginfo

pass-name-in-tablegen

 Pass name in TableGen

StripDebugInfo

Link to this function

strip_debuginfo(composer_or_op)

Link to this function

symbol_dce()

Eliminate dead symbols

argument-name-in-mlir-cli

 Argument name in MLIR CLI

symbol-dce

pass-name-in-tablegen

 Pass name in TableGen

SymbolDCE

Link to this function

symbol_dce(composer_or_op)

Link to this function

symbol_privatize()

Mark symbols private

argument-name-in-mlir-cli

 Argument name in MLIR CLI

symbol-privatize

pass-name-in-tablegen

 Pass name in TableGen

SymbolPrivatize

Link to this function

symbol_privatize(composer_or_op)

Link to this function

topological_sort()

Sort regions without SSA dominance in topological order

argument-name-in-mlir-cli

 Argument name in MLIR CLI

topological-sort

pass-name-in-tablegen

 Pass name in TableGen

TopologicalSort

Link to this function

topological_sort(composer_or_op)

Link to this function

view_op_graph()

Print Graphviz visualization of an operation

argument-name-in-mlir-cli

 Argument name in MLIR CLI

view-op-graph

pass-name-in-tablegen

 Pass name in TableGen

ViewOpGraph

Link to this function

view_op_graph(composer_or_op)