[Mlir-commits] [mlir] [mlir][gpu] Pattern to promote `gpu.shuffle` to specialized AMDGPU ops (PR #137109)

Ivan Butygin llvmlistbot at llvm.org
Sat May 3 01:06:17 PDT 2025


Hardcode84 wrote:

I was thinking about doing it as part of the `gpu.shuffle` lowering. My current motivations to do it separately:
* Easier to test when it's a small separate pattern
* I'm promoting high-level `gpu.shuffle` op to high level `amdgpu.swizzle_bitmode` op. If we decide to do it during lowering, we will need to generate `rocdl` op directly, copypasting part of the exiting shuffle/swizzle lowering. There may also be possible to run further simplifications/promotions on the resulting `amdgpu.swizzle_bitmode` (but I don't have any yet).
* It's opt-in so we won't suddenly break all AMDGPU users if something go wrong

https://github.com/llvm/llvm-project/pull/137109


More information about the Mlir-commits mailing list