[llvm] Reland "[NVPTX][AtomicExpandPass] Complete support for AtomicRMW in NVPTX (#176015)" (PR #179553)

[Artem Belevich via llvm-commits]

Artem-B wrote:

Unfortunately, the test is intertwined with the code I can't share. I'll try to reproduce the issue on a plain CUB test.

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