[llvm] [NVPTX] Use v2.u64 to load/store 128-bit values (PR #136638)
Alex MacLean via llvm-commits
llvm-commits at lists.llvm.org
Tue Apr 22 11:29:22 PDT 2025
AlexMaclean wrote:
> PTX apparently supports 128-bit ld/st/mov on sm_70+ and it looks like we've already landed bits and pieces of support for it last fall, while I wasn't looking. We'll probably need to plumb it through to support 128-bit atomics on sm_90, and it would simplify things here, too (though we'll still need these changes for older GPUs).
>
> @AlexMaclean is that something already on your radar? I was about to start poking at the atomics (#122760), but will be happy if it's about to be done by NVIDIA. :-)
Yea, I've been thinking about improving our 128-bit register support for a while now. I'm hoping to get some time in the next week or two to take a deeper look and clean the situation up a bit.
Sorry to have ignored https://github.com/llvm/llvm-project/issues/122760! I'll look into that too since I'm already trying to do some cleanup around atomics as well.
https://github.com/llvm/llvm-project/pull/136638
More information about the llvm-commits
mailing list