Home
last modified time | relevance | path

Searched refs:GpuAllReduceRewriter (Results 1 – 1 of 1) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/GPU/Transforms/
H A DAllReduceLowering.cpp28 struct GpuAllReduceRewriter { struct
31 GpuAllReduceRewriter(gpu::GPUFuncOp funcOp, gpu::AllReduceOp reduceOp, in GpuAllReduceRewriter() argument
398 GpuAllReduceRewriter(funcOp, reduceOp, rewriter).rewrite(); in matchAndRewrite()