MLIR  20.0.0git
Public Member Functions | List of all members
mlir::transform::gpu::GpuBlockIdBuilder Struct Reference

Builder for gpu::BlockIdOps used to map scf.forall to blocks. More...

#include "mlir/Dialect/GPU/TransformOps/Utils.h"

+ Inheritance diagram for mlir::transform::gpu::GpuBlockIdBuilder:

Public Member Functions

 GpuBlockIdBuilder (MLIRContext *ctx, bool useLinearMapping=false)
 
- Public Member Functions inherited from mlir::transform::gpu::GpuIdBuilder
 GpuIdBuilder ()=default
 
 GpuIdBuilder (MLIRContext *ctx, bool useLinearMapping, const MappingIdBuilderFnType &builder)
 

Additional Inherited Members

- Public Types inherited from mlir::transform::gpu::GpuIdBuilder
using MappingIdBuilderFnType = std::function< DeviceMappingAttrInterface(MLIRContext *, mlir::gpu::MappingId)>
 
- 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...
 

Detailed Description

Builder for gpu::BlockIdOps used to map scf.forall to blocks.

If useLinearMapping is false, the idBuilder method returns 3D values used for indexing rewrites as well as 3D sizes for predicate generation. If useLinearMapping is true, the idBuilder method returns nD values used for indexing rewrites as well as 1D sizes for predicate generation.

Definition at line 83 of file Utils.h.

Constructor & Destructor Documentation

◆ GpuBlockIdBuilder()

mlir::transform::gpu::GpuBlockIdBuilder::GpuBlockIdBuilder ( MLIRContext ctx,
bool  useLinearMapping = false 
)

Definition at line 186 of file Utils.cpp.

References mlir::get().


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