Searched refs:kSubgroupSize (Results 1 – 1 of 1) sorted by relevance
84 create<arith::ConstantIntOp>(kSubgroupSize - 1, int32Type); in rewrite()163 auto bufferType = MemRefType::get({kSubgroupSize}, valueType, AffineMap{}, in createWorkgroupBuffer()325 Value subgroupSize = create<arith::ConstantIntOp>(kSubgroupSize, int32Type); in createSubgroupReduce()338 for (int i = 1; i < kSubgroupSize; i <<= 1) { in createSubgroupReduce()360 for (int i = 1; i < kSubgroupSize; i <<= 1) { in createSubgroupReduce()374 Value subgroupSize = create<arith::ConstantIntOp>(kSubgroupSize, int32Type); in getDivideBySubgroupSize()387 static constexpr int kSubgroupSize = 32; member