2022-06-24 22:28:23 +00:00
|
|
|
struct S_atomic {
|
2022-12-01 17:37:56 +00:00
|
|
|
/* @offset(0) */
|
2022-06-24 22:28:23 +00:00
|
|
|
x : i32,
|
2022-12-01 17:37:56 +00:00
|
|
|
/* @offset(4) */
|
2022-06-24 22:28:23 +00:00
|
|
|
a : atomic<u32>,
|
2022-12-01 17:37:56 +00:00
|
|
|
/* @offset(8) */
|
2022-06-24 22:28:23 +00:00
|
|
|
y : u32,
|
|
|
|
}
|
|
|
|
|
|
|
|
struct S {
|
2022-12-01 17:37:56 +00:00
|
|
|
/* @offset(0) */
|
2022-06-24 22:28:23 +00:00
|
|
|
x : i32,
|
2022-12-01 17:37:56 +00:00
|
|
|
/* @offset(4) */
|
2022-06-24 22:28:23 +00:00
|
|
|
a : u32,
|
2022-12-01 17:37:56 +00:00
|
|
|
/* @offset(8) */
|
2022-06-24 22:28:23 +00:00
|
|
|
y : u32,
|
|
|
|
}
|
|
|
|
|
2023-01-24 14:55:17 +00:00
|
|
|
alias Arr = array<S, 10u>;
|
2022-06-24 22:28:23 +00:00
|
|
|
|
|
|
|
var<private> local_invocation_index_1 : u32;
|
|
|
|
|
|
|
|
var<workgroup> wg : array<S_atomic, 10u>;
|
|
|
|
|
2023-02-21 17:33:54 +00:00
|
|
|
fn compute_main_inner(local_invocation_index_2 : u32) {
|
2023-05-23 18:28:44 +00:00
|
|
|
var idx = 0u;
|
2023-02-21 17:33:54 +00:00
|
|
|
idx = local_invocation_index_2;
|
2022-06-24 22:28:23 +00:00
|
|
|
loop {
|
2023-05-23 18:28:44 +00:00
|
|
|
let x_23 = idx;
|
2022-06-24 22:28:23 +00:00
|
|
|
if (!((x_23 < 10u))) {
|
|
|
|
break;
|
|
|
|
}
|
2023-05-23 18:28:44 +00:00
|
|
|
let x_28 = idx;
|
2022-06-24 22:28:23 +00:00
|
|
|
wg[x_28].x = 0i;
|
|
|
|
atomicStore(&(wg[x_28].a), 0u);
|
|
|
|
wg[x_28].y = 0u;
|
|
|
|
|
|
|
|
continuing {
|
2023-05-23 18:28:44 +00:00
|
|
|
let x_41 = idx;
|
2022-06-24 22:28:23 +00:00
|
|
|
idx = (x_41 + 1u);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
workgroupBarrier();
|
|
|
|
atomicStore(&(wg[4i].a), 1u);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
fn compute_main_1() {
|
2023-05-23 18:28:44 +00:00
|
|
|
let x_53 = local_invocation_index_1;
|
2022-06-24 22:28:23 +00:00
|
|
|
compute_main_inner(x_53);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
@compute @workgroup_size(1i, 1i, 1i)
|
|
|
|
fn compute_main(@builtin(local_invocation_index) local_invocation_index_1_param : u32) {
|
|
|
|
local_invocation_index_1 = local_invocation_index_1_param;
|
|
|
|
compute_main_1();
|
|
|
|
}
|