MLIR
22.0.0git
|
#include "mlir/Dialect/Linalg/Transforms/Transforms.h"
Public Attributes | |
SmallVector< int64_t, 3 > | blockFactors |
Minor block factors (mb, nb, kb) for packing relayout where mb, mn are the parallel dimensions and kb is the reduction dimension. More... | |
bool | allowPadding = true |
If true, allows packing of dimensions that only partially fit into the block factors. More... | |
SmallVector< int64_t, 3 > | mnkPaddedSizesNextMultipleOf |
Next multiples of the packing sizes. More... | |
SmallVector< int64_t, 3 > | mnkOrder = {0, 1, 2} |
Permutation of matmul (M, N, K) dimensions order. More... | |
bool | lhsTransposeOuterBlocks = false |
Transpose LHS outer block layout [MB][KB] -> [KB][MB]. More... | |
bool | lhsTransposeInnerBlocks = false |
Transpose LHS inner block layout [mb][kb] -> [kb][mb]. More... | |
bool | rhsTransposeOuterBlocks = true |
Transpose RHS outer block layout [KB][NB] -> [NB][KB]. More... | |
bool | rhsTransposeInnerBlocks = true |
Transpose RHS inner block layout [kb][nb] -> [nb][kb]. More... | |
Definition at line 1306 of file Transforms.h.
bool mlir::linalg::BlockPackMatmulOptions::allowPadding = true |
If true, allows packing of dimensions that only partially fit into the block factors.
Definition at line 1313 of file Transforms.h.
SmallVector<int64_t, 3> mlir::linalg::BlockPackMatmulOptions::blockFactors |
Minor block factors (mb, nb, kb) for packing relayout where mb, mn are the parallel dimensions and kb is the reduction dimension.
Definition at line 1309 of file Transforms.h.
bool mlir::linalg::BlockPackMatmulOptions::lhsTransposeInnerBlocks = false |
Transpose LHS inner block layout [mb][kb] -> [kb][mb].
Definition at line 1325 of file Transforms.h.
bool mlir::linalg::BlockPackMatmulOptions::lhsTransposeOuterBlocks = false |
Transpose LHS outer block layout [MB][KB] -> [KB][MB].
Definition at line 1322 of file Transforms.h.
SmallVector<int64_t, 3> mlir::linalg::BlockPackMatmulOptions::mnkOrder = {0, 1, 2} |
Permutation of matmul (M, N, K) dimensions order.
Definition at line 1319 of file Transforms.h.
SmallVector<int64_t, 3> mlir::linalg::BlockPackMatmulOptions::mnkPaddedSizesNextMultipleOf |
Next multiples of the packing sizes.
Definition at line 1316 of file Transforms.h.
bool mlir::linalg::BlockPackMatmulOptions::rhsTransposeInnerBlocks = true |
Transpose RHS inner block layout [kb][nb] -> [nb][kb].
Definition at line 1331 of file Transforms.h.
bool mlir::linalg::BlockPackMatmulOptions::rhsTransposeOuterBlocks = true |
Transpose RHS outer block layout [KB][NB] -> [NB][KB].
Definition at line 1328 of file Transforms.h.