MLIR 22.0.0git
Transforms.h File Reference
#include "mlir/IR/Operation.h"
#include "llvm/ADT/SmallVector.h"
#include "llvm/Support/LogicalResult.h"
#include <functional>
#include <optional>
#include <utility>

Go to the source code of this file.

Classes

struct  mlir::xegpu::UnrollOptions
 Options to control the XeGPU unrolling. More...

Namespaces

namespace  mlir
 Include the generated interface declarations.
namespace  mlir::xegpu

Functions

void mlir::xegpu::populateXeGPUFoldAliasOpsPatterns (RewritePatternSet &patterns)
 Appends patterns for folding aliasing ops into XeGPU ops into patterns.
void mlir::xegpu::populateXeGPUOptimizeBlockLoadsPatterns (RewritePatternSet &patterns)
 Appends patterns for optimizing block load operations into patterns.
void mlir::xegpu::populateXeGPUSubgroupDistributePatterns (RewritePatternSet &patterns)
 Appends patterns for XeGPU SIMT distribution into patterns.
void mlir::xegpu::populateXeGPUMoveFuncBodyToWarpOpPatterns (RewritePatternSet &patterns)
 Appends patterns for moving function body into gpu.warp_execute_on_lane0 op.
void mlir::xegpu::populateXeGPUWgToSgDistributePatterns (RewritePatternSet &patterns)
 Appends patterns for XeGPU workgroup to subgroup distribution into patterns.
void mlir::xegpu::populateXeGPUUnrollPatterns (RewritePatternSet &patterns, const UnrollOptions &options)
 Collect a set of patterns to unroll xegpu operations to a smaller shapes.