blob: 95af29c38cb7ac39a1d739e064709b4e479a2ad3 [file] [log] [blame] [edit]
// SPDX-License-Identifier: Apache-2.0 OR MIT OR Unlicense
// Set up dispatch size for path count stage.
#import bump
@group(0) @binding(0)
var<storage> bump: BumpAllocators;
@group(0) @binding(1)
var<storage, read_write> indirect: IndirectCount;
// Partition size for path count stage
let WG_SIZE = 256u;
@compute @workgroup_size(1)
fn main() {
let lines = atomicLoad(&bump.lines);
indirect.count_x = (lines + (WG_SIZE - 1u)) / WG_SIZE;
indirect.count_y = 1u;
indirect.count_z = 1u;
}