Home
last modified time | relevance | path

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

/llvm-project-15.0.7/mlir/lib/Dialect/GPU/Transforms/
H A DAllReduceLowering.cpp79 Value invocationIdx = create<arith::AddIOp>(int32Type, tmp3, tidX); in rewrite() local
85 Value laneId = create<arith::AndIOp>(invocationIdx, subgroupMask); in rewrite()
91 create<arith::SubIOp>(invocationIdx, laneId); in rewrite()
112 Value subgroupId = getDivideBySubgroupSize(invocationIdx); in rewrite()
123 invocationIdx, numSubgroups); in rewrite()
130 Value index = create<arith::IndexCastOp>(indexType, invocationIdx); in rewrite()