[WebGPU] Control Access Barriers

[cmhhelgeson]

Description

At some point, I think it would be great to have the ability to set control access barriers and access more of the WGSL specific functionality within a TSL function. I've tried implementing some of these access barriers myself in a personal branch, using an approach similar to that of the 'vec2', 'vec3', 'etc' ConvertTypes in ShaderNode.js. However, it seems like the node system is heavily weighted towards nodes that explicitly represent some kind of mathematical operation, so, at least from my end, it's not as simple as just adding another node type.

Solution

export const workgroupBarrier = new BarrierType('workgroup');

And then be accessed from a tslFn like so...

tslFn( () => {
	...
	workgroupBarrier()
	...
})

I'm not sure if there's currently a relevant use case for these barriers, but once subgroups become stable, I imagine commands like workgroupBarrier() or subgroupBarrier() would be very useful for things like prefix sums for spatial hash grids.

Alternatives

No response

Additional context

No response

[sunag]

I think the code below can work for a tslFn() using as your example?

export const workgroupBarrier = () => expression( 'workgroupBarrier()' ).append();