MLIR  20.0.0git
Classes | Namespaces | Functions
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

 mlir
 Include the generated interface declarations.
 

Functions

bool mlir::isRowMajorMatmul (ArrayAttr indexingMaps)
 Tests whether the given maps describe a row major matmul. More...
 
bool mlir::isColumnMajorMatmul (ArrayAttr indexingMaps)
 Tests whether the given maps describe a column major matmul. More...
 
bool mlir::isRowMajorBatchMatmul (ArrayAttr indexingMaps)
 Tests whether the given maps describe a row major batch matmul. More...
 
bool mlir::isVecmat (ArrayAttr indexingMaps)
 Tests whether the given maps describe a vector matrix multiplication. More...
 
bool mlir::isBatchVecmat (ArrayAttr indexingMaps)
 Tests whether the given maps describe a batch vector matrix multiplication. More...
 
bool mlir::isMatvec (ArrayAttr indexingMaps)
 Tests whether the given maps describe a matrix vector multiplication. More...
 
bool mlir::isBatchMatvec (ArrayAttr indexingMaps)
 Tests whether the given maps describe a batch matrix vector multiplication. More...
 
void mlir::findPositionsOfType (ArrayRef< utils::IteratorType > iteratorTypes, utils::IteratorType iteratorTypeName, SmallVectorImpl< unsigned > &res)
 Return positions in iteratorTypes that match iteratorTypeName. More...
 
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)