Added NodeJS Demo.
This commit is contained in:
68
demos/sort/shaders/bitonicSortUIntMultiPass.wgsl
Normal file
68
demos/sort/shaders/bitonicSortUIntMultiPass.wgsl
Normal file
@@ -0,0 +1,68 @@
|
||||
@group(0) @binding(0)
|
||||
var<storage, read_write> compare: array<u32>;
|
||||
|
||||
@group(0) @binding(1)
|
||||
var<storage, read_write> threadPassIndices: array<u32>;
|
||||
|
||||
@group(0) @binding(2)
|
||||
var<storage, read_write> kArray: array<u32>;
|
||||
|
||||
@group(0) @binding(3)
|
||||
var<storage, read_write> jArray: array<u32>;
|
||||
|
||||
@group(0) @binding(4)
|
||||
var<uniform> totalCount: u32;
|
||||
|
||||
|
||||
|
||||
@compute @workgroup_size( 256 )
|
||||
fn main( @builtin(global_invocation_id) global_id : vec3<u32> ) {
|
||||
|
||||
let idx = global_id.x;
|
||||
|
||||
let threadPassIndex = threadPassIndices[ idx ];
|
||||
|
||||
threadPassIndices[idx] = threadPassIndices[idx] + 1;
|
||||
|
||||
let j = jArray[ threadPassIndex ];
|
||||
|
||||
let k = kArray[ threadPassIndex ];
|
||||
|
||||
let ixj = idx ^ j;
|
||||
|
||||
|
||||
|
||||
if (idx >= totalCount || ixj <= idx || ixj >= totalCount) {
|
||||
return;
|
||||
}
|
||||
|
||||
if (ixj > idx) {
|
||||
let ascending = (idx & k) == 0u;
|
||||
|
||||
let dist_idx = compare[idx];
|
||||
let dist_ixj = compare[ixj];
|
||||
|
||||
var swap = false;
|
||||
|
||||
if (ascending) {
|
||||
if (dist_idx > dist_ixj) {
|
||||
swap = true;
|
||||
}
|
||||
} else {
|
||||
if (dist_idx < dist_ixj) {
|
||||
swap = true;
|
||||
}
|
||||
}
|
||||
|
||||
if (swap) {
|
||||
|
||||
let tempDist = compare[idx];
|
||||
|
||||
compare[idx] = compare[ixj];
|
||||
compare[ixj] = tempDist;
|
||||
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
}
|
||||
Reference in New Issue
Block a user