https://github.com/krzysz00 approved this pull request. Looks good to me! I think, long-term, we'll need some way to map GPU memory spaces to their LLVM counterparts in an IR-visible way, but that's not this PR. https://github.com/llvm/llvm-project/pull/77367