Helper struct for configuring the rewrite of mapped scf.forall ops to various gpu id configurations. More...
#include "mlir/Dialect/GPU/TransformOps/Utils.h"
Inheritance diagram for mlir::transform::gpu::GpuIdBuilder:
Public Types | |
| using | MappingIdBuilderFnType = std::function< DeviceMappingAttrInterface(MLIRContext *, mlir::gpu::MappingId)> |
Public Member Functions | |
| GpuIdBuilder ()=default | |
| GpuIdBuilder (MLIRContext *ctx, bool useLinearMapping, const MappingIdBuilderFnType &builder) | |
Public Attributes | |
| SmallVector< DeviceMappingAttrInterface > | mappingAttributes |
| The mapping attributes targeted by this generator. More... | |
| GpuIdBuilderFnType | idBuilder |
| The constructor that builds the concrete IR for mapping ids. More... | |
Detailed Description
Helper struct for configuring the rewrite of mapped scf.forall ops to various gpu id configurations.
Member Typedef Documentation
◆ MappingIdBuilderFnType
| using mlir::transform::gpu::GpuIdBuilder::MappingIdBuilderFnType = std::function<DeviceMappingAttrInterface( MLIRContext *, mlir::gpu::MappingId)> |
Constructor & Destructor Documentation
◆ GpuIdBuilder() [1/2]
|
default |
◆ GpuIdBuilder() [2/2]
| mlir::transform::gpu::GpuIdBuilder::GpuIdBuilder | ( | MLIRContext * | ctx, |
| bool | useLinearMapping, | ||
| const MappingIdBuilderFnType & | builder | ||
| ) |
Definition at line 171 of file Utils.cpp.
References mappingAttributes.
Member Data Documentation
◆ idBuilder
| GpuIdBuilderFnType mlir::transform::gpu::GpuIdBuilder::idBuilder |
The constructor that builds the concrete IR for mapping ids.
Definition at line 75 of file Utils.h.
Referenced by rewriteOneForallCommonImpl().
◆ mappingAttributes
| SmallVector<DeviceMappingAttrInterface> mlir::transform::gpu::GpuIdBuilder::mappingAttributes |
The mapping attributes targeted by this generator.
Definition at line 72 of file Utils.h.
Referenced by GpuIdBuilder(), and rewriteOneForallCommonImpl().
The documentation for this struct was generated from the following files:
