MLIR 22.0.0git
mlir::transform::gpu::GpuLaneIdBuilder Struct Reference

Builder for lane id. More...

#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.
GpuIdBuilderFnType idBuilder
 The constructor that builds the concrete IR for mapping ids.

Additional Inherited Members

Public Types inherited from mlir::transform::gpu::GpuIdBuilder
using MappingIdBuilderFnType

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.

Definition at line 130 of file Utils.h.

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::transform::gpu::GpuIdBuilder::GpuIdBuilder(), true, and warpSize.

Member Data Documentation

◆ warpSize

int64_t mlir::transform::gpu::GpuLaneIdBuilder::warpSize = 32

Definition at line 133 of file Utils.h.

Referenced by GpuLaneIdBuilder().


The documentation for this struct was generated from the following files: