MLIR 22.0.0git
GPUAllReduceConversion Class Referencefinal

Pattern to convert a gpu.all_reduce op into a SPIR-V group op. More...

Inheritance diagram for GPUAllReduceConversion:

Public Member Functions

LogicalResult matchAndRewrite (gpu::AllReduceOp op, OpAdaptor adaptor, ConversionPatternRewriter &rewriter) const override

Detailed Description

Pattern to convert a gpu.all_reduce op into a SPIR-V group op.

Definition at line 653 of file GPUToSPIRV.cpp.

Member Function Documentation

◆ matchAndRewrite()

LogicalResult GPUAllReduceConversion::matchAndRewrite ( gpu::AllReduceOp op,
OpAdaptor adaptor,
ConversionPatternRewriter & rewriter ) const
inlineoverride

Definition at line 659 of file GPUToSPIRV.cpp.

References createGroupReduceOp(), result, and success().


The documentation for this class was generated from the following file: