[mlir][Transform] Fix support for mapping to GPU warps and to linear ids
c59465e1 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.
Loading
Please sign in to comment