Home
last modified time | relevance | path

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

/llvm-project-15.0.7/mlir/lib/Dialect/GPU/Transforms/
H A DAllReduceLowering.cpp325 Value subgroupSize = create<arith::ConstantIntOp>(kSubgroupSize, int32Type); in createSubgroupReduce() local
327 activeWidth, subgroupSize); in createSubgroupReduce()
363 create<gpu::ShuffleOp>(shuffleType, value, offset, subgroupSize, in createSubgroupReduce()
374 Value subgroupSize = create<arith::ConstantIntOp>(kSubgroupSize, int32Type); in getDivideBySubgroupSize() local
375 return create<arith::DivSIOp>(int32Type, value, subgroupSize); in getDivideBySubgroupSize()