[mlir][gpu] Improve foreach_thread distribution
authorThomas Raoux <thomasraoux@google.com>
Fri, 13 Jan 2023 19:53:56 +0000 (19:53 +0000)
committerThomas Raoux <thomasraoux@google.com>
Tue, 17 Jan 2023 17:12:55 +0000 (17:12 +0000)
commit794979ad8cc975b4417dcf5fcce9c9e8e28e52b3
treec039a7b4350c44b51716fd666bf5d753c713815f
parent435905ecf25ab9da0753931358414164352810f5
[mlir][gpu] Improve foreach_thread distribution

Replace Ids with 0 when block dim is 1 when distributing foreach_thread.

Differential Revision: https://reviews.llvm.org/D141718
mlir/lib/Dialect/GPU/TransformOps/GPUTransformOps.cpp
mlir/test/Dialect/GPU/transform-gpu.mlir