Home
last modified time | relevance | path

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

/llvm-project-15.0.7/mlir/lib/Dialect/GPU/Transforms/
H A DAllReduceLowering.cpp121 Value numSubgroups = getDivideBySubgroupSize(biasedBlockSize); in rewrite() local
123 invocationIdx, numSubgroups); in rewrite()
133 createSubgroupReduce(numSubgroups, laneId, value, accumFactory); in rewrite()