| [MLIR] Create GPU utils library & move distribution utils (#119264) Continue the move of warp_execute_on_lane_0 op to the gpu dialect (#116994). This patch creates a utils library in GPU and moves generic helper functions there. | 1 年前 |
| [mlir] Use llvm accumulate wrappers. NFCI. (#162957) Use wrappers around std::accumulate to make the code more concise and less bug-prone: https://github.com/llvm/llvm-project/pull/162129. With std::accumulate, it's the initial value that determines the accumulator type. llvm::sum_of and llvm::product_of pick the right accumulator type based on the range element type. Found some funny bugs like a local accumulate helper that calculated a sum with initial value of 1 -- we didn't hit the bug because the code was actually dead... | 10 个月前 |
| [MLIR] Create GPU utils library & move distribution utils (#119264) Continue the move of warp_execute_on_lane_0 op to the gpu dialect (#116994). This patch creates a utils library in GPU and moves generic helper functions there. | 1 年前 |