@@ -141,20 +141,21 @@ void dispatch_bool_op(
141141 graph.add_dispatch ({pipeline, bind_group, workgroup_count});
142142
143143 WGPUBuffer p_buf = params_buf;
144- auto resize = [self_id, out_id, scalar, wg_size, dispatch_idx, p_buf, op_name](
145- WebGPUGraph& g) {
146- const auto & d = g.cur_dims (self_id);
147- uint32_t n = 1u ;
148- for (auto x : d) {
149- n *= static_cast <uint32_t >(x);
150- }
151- g.set_cur_dims (out_id, d);
152- BoolOpParams p = {n, scalar, 0u , 0u };
153- wgpuQueueWriteBuffer (g.queue (), p_buf, 0 , &p, sizeof (p));
154- const uint32_t nw = (n + 3u ) / 4u ;
155- g.dispatch_at (dispatch_idx).workgroup_count_x =
156- utils::compute_1d_workgroup_count (g.device (), nw, wg_size, op_name);
157- };
144+ auto resize =
145+ [self_id, out_id, scalar, wg_size, dispatch_idx, p_buf, op_name](
146+ WebGPUGraph& g) {
147+ const auto & d = g.cur_dims (self_id);
148+ uint32_t n = 1u ;
149+ for (auto x : d) {
150+ n *= static_cast <uint32_t >(x);
151+ }
152+ g.set_cur_dims (out_id, d);
153+ BoolOpParams p = {n, scalar, 0u , 0u };
154+ wgpuQueueWriteBuffer (g.queue (), p_buf, 0 , &p, sizeof (p));
155+ const uint32_t nw = (n + 3u ) / 4u ;
156+ g.dispatch_at (dispatch_idx).workgroup_count_x =
157+ utils::compute_1d_workgroup_count (g.device (), nw, wg_size, op_name);
158+ };
158159 graph.add_tensor_resize_hook (self_id, resize);
159160
160161 wgpuShaderModuleRelease (shader);
0 commit comments