Make use of newly added subgroups

[alienself]

Hi,

Quick question, since Chrome 128, webgpu now has experimental subgroups, would it be possible to add native support for them using the currently available WGSL API?

See: https://developer.chrome.com/blog/new-in-webgpu-128#experimenting_with_subgroups

It is still behind the webgpu flag: chrome://flags/#enable-unsafe-webgpu (or by signing up for the origin trial) for the next 4 versions of chrome. All the subgroup functions mentioned in the proposal should be available.

It would be amazing to finally have a super fast and optimized sorting solution with webgpu!

Thanks!

[tombh]

I assume you've seen this? It has a subgroup-based implementation using WGSL: https://github.com/b0nes164/GPUPrefixSums/blob/main/GPUPrefixSumsWGPU/src/Shaders/csdl.wgsl

[alienself]

Hi! Thanks for the reply, yes I saw it but this is on his GPUPrefixSum repo which means that it's missing the sorting part. I also opened an issue on his GPU sorting repo (b0nes164/GPUSorting#6) to which he commented however it seems like this is not going to happen in the immediate future.
Did you have a look at his prefix sum implementation with subgroups?

[tombh]

I had a look, but I couldn't use it because I'm on Asahi Linux (M1 Mac) and the OpenGL driver doesn't support subgroups. I'm waiting for Vulkan support so I can properly try out subgroups.