Searched refs:shuffleType (Results 1 – 1 of 1) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Dialect/GPU/Transforms/ |
| H A D | AllReduceLowering.cpp | 328 std::array<Type, 2> shuffleType = {valueType, rewriter.getI1Type()}; in createSubgroupReduce() local
341 shuffleType, value, offset, activeWidth, gpu::ShuffleMode::XOR); in createSubgroupReduce()
363 create<gpu::ShuffleOp>(shuffleType, value, offset, subgroupSize, in createSubgroupReduce()
|