MLIR  22.0.0git
Namespaces | Functions
Transforms.h File Reference
#include "mlir/Dialect/Shard/IR/ShardOps.h"
#include "mlir/IR/BuiltinTypes.h"
#include "mlir/IR/Value.h"
#include "mlir/Support/LLVM.h"
#include "llvm/ADT/ArrayRef.h"

Go to the source code of this file.

Namespaces

 mlir
 Include the generated interface declarations.
 
 mlir::shard
 

Functions

void mlir::shard::populateProcessMultiIndexOpLoweringPatterns (RewritePatternSet &patterns, SymbolTableCollection &symbolTableCollection)
 
void mlir::shard::registerProcessMultiIndexOpLoweringDialects (DialectRegistry &registry)
 
void mlir::shard::populateAllSliceOpLoweringPatterns (RewritePatternSet &patterns, SymbolTableCollection &symbolTableCollection)
 
void mlir::shard::registerAllSliceOpLoweringDialects (DialectRegistry &registry)
 
void mlir::shard::populateAllOpLoweringPatterns (RewritePatternSet &patterns, SymbolTableCollection &symbolTableCollection)
 
void mlir::shard::registerAllOpLoweringDialects (DialectRegistry &registry)
 
TypedValue< IndexType > mlir::shard::createCollectiveProcessGroupSize (GridOp grid, ArrayRef< GridAxis > axes, ImplicitLocOpBuilder &builder)
 
TypedValue< IndexType > mlir::shard::createProcessLinearIndex (StringRef grid, ArrayRef< GridAxis > gridAxes, ImplicitLocOpBuilder &builder)
 
TypedValue< IndexType > mlir::shard::createProcessLinearIndex (StringRef grid, ValueRange processInGroupMultiIndex, ArrayRef< GridAxis > gridAxes, ImplicitLocOpBuilder &builder)