Home
last modified time | relevance | path

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

/llvm-project-15.0.7/mlir/lib/Dialect/GPU/Transforms/
H A DAllReduceLowering.cpp83 Value subgroupMask = in rewrite() local
85 Value laneId = create<arith::AndIOp>(invocationIdx, subgroupMask); in rewrite()
120 create<arith::AddIOp>(int32Type, workgroupSize, subgroupMask); in rewrite()