[mlir][Transform] Fix support for mapping to GPU warps and to linear ids
authorNicolas Vasilache <nicolas.vasilache@gmail.com>
Mon, 20 Mar 2023 12:17:39 +0000 (05:17 -0700)
committerNicolas Vasilache <nicolas.vasilache@gmail.com>
Mon, 20 Mar 2023 12:23:17 +0000 (05:23 -0700)
commitba7f3e1d1e50212bdc8cc438185519fd7257aa44
treeecc91c06b595718c20ca74763eaa2b75d230df59
parent0107513fe79da7670e37c29c0862794a2213a89c
[mlir][Transform] Fix support for mapping to GPU warps and to linear ids

c59465e1203dd78d06e15f7ddf62141807dbd5a7 introduced mapping to warps and
linear GPU ids.

In the implementation, the delinearization basis is reversed from [x, y, z]
to [z, y x] order to properly compute the strides and allow delinearization.

Prior to this commit, we forgot to reverse it back to [x, y, z] order
before materializing the indices.

Fix this oversight.
mlir/lib/Dialect/GPU/TransformOps/GPUTransformOps.cpp
mlir/test/Dialect/GPU/transform-gpu.mlir