// SPDX-License-Identifier: Apache-2.0 OR MIT OR Unlicense

// Set up dispatch size for path tiling stage.

#import bump

@group(0) @binding(0)
var<storage, read_write> bump: BumpAllocators;

@group(0) @binding(1)
var<storage, read_write> indirect: IndirectCount;

// Partition size for path tiling stage
let WG_SIZE = 256u;

@compute @workgroup_size(1)
fn main() {
    let segments = atomicLoad(&bump.seg_counts);
    indirect.count_x = (segments + (WG_SIZE - 1u)) / WG_SIZE;
    indirect.count_y = 1u;
    indirect.count_z = 1u;
}