#include "mlir/Dialect/GPU/TransformOps/Utils.h"
Inheritance diagram for mlir::transform::gpu::GpuLaneIdBuilder:Public Member Functions | |
| GpuLaneIdBuilder (MLIRContext *ctx, int64_t warpSize, bool unused, DeviceMaskingAttrInterface mask=nullptr) | |
| Public Member Functions inherited from mlir::transform::gpu::GpuIdBuilder | |
| GpuIdBuilder ()=default | |
| GpuIdBuilder (MLIRContext *ctx, bool useLinearMapping, const MappingIdBuilderFnType &builder) | |
Public Attributes | |
| int64_t | warpSize = 32 |
| Public Attributes inherited from mlir::transform::gpu::GpuIdBuilder | |
| SmallVector< DeviceMappingAttrInterface > | mappingAttributes |
| The mapping attributes targeted by this generator. More... | |
| GpuIdBuilderFnType | idBuilder |
| The constructor that builds the concrete IR for mapping ids. More... | |
Additional Inherited Members | |
| Public Types inherited from mlir::transform::gpu::GpuIdBuilder | |
| using | MappingIdBuilderFnType = std::function< DeviceMappingAttrInterface(MLIRContext *, mlir::gpu::MappingId)> |
Detailed Description
Builder for lane id.
The idBuilder method returns nD values used for indexing rewrites as well as 1D sizes for predicate generation. This useLinearMapping case is the only supported case. If mask is provided, it will be used to filter the active lanes.
Constructor & Destructor Documentation
◆ GpuLaneIdBuilder()
| mlir::transform::gpu::GpuLaneIdBuilder::GpuLaneIdBuilder | ( | MLIRContext * | ctx, |
| int64_t | warpSize, | ||
| bool | unused, | ||
| DeviceMaskingAttrInterface | mask = nullptr |
||
| ) |
Definition at line 349 of file Utils.cpp.
References mlir::get().
Member Data Documentation
◆ warpSize
| int64_t mlir::transform::gpu::GpuLaneIdBuilder::warpSize = 32 |
The documentation for this struct was generated from the following files:
