GPUTransformOps.h File Reference

#include "mlir/Dialect/SCF/IR/SCF.h"
#include "mlir/Dialect/Transform/Interfaces/TransformInterfaces.h"
#include "mlir/IR/PatternMatch.h"
#include "mlir/Dialect/GPU/TransformOps/GPUTransformOps.h.inc"

Go to the source code of this file.

Namespaces
namespace	mlir
	Include the generated interface declarations.
namespace	mlir::gpu
namespace	mlir::transform
namespace	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.
DiagnosedSilenceableFailure	mlir::transform::gpu::mapOneForallToThreadsImpl (RewriterBase &rewriter, std::optional< TransformOpInterface > transformOp, scf::ForallOp forallOp, ArrayRef< int64_t > blockSizes, int64_t warpSize, bool syncAfterDistribute)
	Search scf.forall ops nested under target and map each such op to an explicit GPU implementation along blockDims.
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.
void	mlir::gpu::registerTransformDialectExtension (DialectRegistry &registry)

Macro Definition Documentation

GET_OP_CLASSES

#define GET_OP_CLASSES

Definition at line 26 of file GPUTransformOps.h.