MLIR  20.0.0git
Namespaces | Macros | Functions
GPUTransformOps.h File Reference
#include "mlir/Dialect/SCF/IR/SCF.h"
#include "mlir/Dialect/Transform/Interfaces/TransformInterfaces.h"
#include "mlir/IR/OpImplementation.h"
#include "mlir/IR/PatternMatch.h"
#include "mlir/Dialect/GPU/TransformOps/GPUTransformOps.h.inc"

Go to the source code of this file.

Namespaces

 mlir
 Include the generated interface declarations.
 
 mlir::gpu
 
 mlir::transform
 
 mlir::transform::gpu
 

Macros

#define GET_OP_CLASSES
 

Functions

DiagnosedSilenceableFailure mlir::transform::gpu::mapForallToBlocksImpl (RewriterBase &rewriter, TransformOpInterface transformOp, scf::ForallOp forallOp, SmallVectorImpl< int64_t > &gridDims, const GpuIdBuilder &gpuIdBuilder)
 Map the top level scf.forall op to GPU blocks. More...
 
DiagnosedSilenceableFailure mlir::transform::gpu::mapOneForallToThreadsImpl (RewriterBase &rewriter, std::optional< TransformOpInterface > transformOp, scf::ForallOp forallOp, ArrayRef< int64_t > blockDims, int64_t warpSize, bool syncAfterDistribute)
 Search scf.forall ops nested under target and map each such op to an explicit GPU implementation along blockDims. More...
 
DiagnosedSilenceableFailure mlir::transform::gpu::mapNestedForallToThreadsImpl (RewriterBase &rewriter, std::optional< TransformOpInterface > transformOp, Operation *target, ArrayRef< int64_t > blockDims, int64_t warpSize, bool syncAfterDistribute)
 Search scf.forall ops nested under target and map each such op to an explicit GPU implementation along blockDims. More...
 
void mlir::gpu::registerTransformDialectExtension (DialectRegistry &registry)
 

Macro Definition Documentation

◆ GET_OP_CLASSES

#define GET_OP_CLASSES

Definition at line 27 of file GPUTransformOps.h.