[mlir][nvgpu] Improve finding module Op to for `mbarrier.create`
authorGuray Ozen <guray.ozen@gmail.com>
Fri, 21 Jul 2023 08:36:29 +0000 (10:36 +0200)
committerGuray Ozen <guray.ozen@gmail.com>
Fri, 21 Jul 2023 08:36:45 +0000 (10:36 +0200)
commit9dad32cb9094e3e3da5d30bfe2aa2d5cf2b066a5
treeebcb80e703ae1eb0aea50bfa59c6509c47c366ef
parent70c2e0618a0f3c09ed7149d88b4987b932eb6705
[mlir][nvgpu] Improve finding module Op to for `mbarrier.create`

Current transformation expects module op to be two level higher, however, it is not always the case. This work searches module op in a while loop.

Reviewed By: nicolasvasilache

Differential Revision: https://reviews.llvm.org/D155825
mlir/lib/Conversion/NVGPUToNVVM/NVGPUToNVVM.cpp
mlir/test/Conversion/NVGPUToNVVM/nvgpu-to-nvvm.mlir