playcanvas

Version:
Open-source WebGL/WebGPU 3D engine for the web
35 lines (34 loc) • 932 B
JavaScript
const computeGsplatLocalDispatchPrepSource = `
@group(0) @binding(0) var<storage, read> sortElementCount: array<u32>;
@group(0) @binding(1) var<storage, read_write> dispatchArgs: array<u32>;
@compute @workgroup_size(1)
fn main() {
	let count = sortElementCount[0];
	let maxDim = {MAX_DIM}u;
	let countWg = (count + {SPLATS_PER_WG_MINUS_1}u) / {SPLATS_PER_WG}u;
	if (countWg <= maxDim) {
		dispatchArgs[0] = countWg;
		dispatchArgs[1] = 1u;
	} else {
		let y = (countWg + maxDim - 1u) / maxDim;
		let x = (countWg + y - 1u) / y;
		dispatchArgs[0] = x;
		dispatchArgs[1] = y;
	}
	dispatchArgs[2] = 1u;
	let placeWg = (count + 255u) / 256u;
	if (placeWg <= maxDim) {
		dispatchArgs[3] = placeWg;
		dispatchArgs[4] = 1u;
	} else {
		let y = (placeWg + maxDim - 1u) / maxDim;
		let x = (placeWg + y - 1u) / y;
		dispatchArgs[3] = x;
		dispatchArgs[4] = y;
	}
	dispatchArgs[5] = 1u;
}
`;
export {
	computeGsplatLocalDispatchPrepSource
};