MLIR 22.0.0git
StructuredOpsUtils.h File Reference
#include "mlir/IR/AffineMap.h"
#include "mlir/IR/BuiltinAttributes.h"
#include "mlir/IR/Location.h"
#include "mlir/IR/TypeRange.h"
#include "mlir/Support/LLVM.h"
#include "mlir/Dialect/Utils/DialectUtilsEnums.h.inc"

Go to the source code of this file.

Classes

class  mlir::StructuredGenerator< StructuredOpInterface, IteratorTypeT >
 Helper StructuredGenerator class to manipulate and rewrite ops with StructuredOpInterface. More...
struct  mlir::StructuredGenerator< StructuredOpInterface, IteratorTypeT >::IteratorType
struct  mlir::StructuredGenerator< StructuredOpInterface, IteratorTypeT >::Par
struct  mlir::StructuredGenerator< StructuredOpInterface, IteratorTypeT >::Red

Namespaces

namespace  mlir
 Include the generated interface declarations.

Functions

bool mlir::isRowMajorMatmul (ArrayAttr indexingMaps)
 Tests whether the given maps describe a row major matmul.
bool mlir::isColumnMajorMatmul (ArrayAttr indexingMaps)
 Tests whether the given maps describe a column major matmul.
bool mlir::isRowMajorBatchMatmul (ArrayAttr indexingMaps)
 Tests whether the given maps describe a row major batch matmul.
bool mlir::isVecmat (ArrayAttr indexingMaps)
 Tests whether the given maps describe a vector matrix multiplication.
bool mlir::isBatchVecmat (ArrayAttr indexingMaps)
 Tests whether the given maps describe a batch vector matrix multiplication.
bool mlir::isMatvec (ArrayAttr indexingMaps)
 Tests whether the given maps describe a matrix vector multiplication.
bool mlir::isBatchMatvec (ArrayAttr indexingMaps)
 Tests whether the given maps describe a batch matrix vector multiplication.
void mlir::findPositionsOfType (ArrayRef< utils::IteratorType > iteratorTypes, utils::IteratorType iteratorTypeName, SmallVectorImpl< unsigned > &res)
 Return positions in iteratorTypes that match iteratorTypeName.
Operationmlir::clone (OpBuilder &b, Operation *op, TypeRange newResultTypes, ValueRange newOperands)
template<typename OpT>
OpT mlir::clone (OpBuilder &b, OpT op, TypeRange newResultTypes, ValueRange newOperands)
Operationmlir::cloneWithoutRegions (OpBuilder &b, Operation *op, TypeRange newResultTypes, ValueRange newOperands)
SmallVector< NamedAttributemlir::getPrunedAttributeList (Operation *op, ArrayRef< StringRef > elidedAttrs)