adrien/zig-wgpu
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Removed TensorInfo from example shader

Browse Source
This commit is contained in:
adrien 2026-05-18 16:45:55 +02:00
parent c62bead6e9
commit 3d9cb4df04
1 changed files with 2 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/shaders/add.wgsl
View File

@ -3,11 +3,7 @@ enable f16;
@group(0) @binding(0) var<storage, read> A: array<f16>; @group(0) @binding(0) var<storage, read> A: array<f16>;
@group(0) @binding(1) var<storage, read> B: array<f16>; @group(0) @binding(1) var<storage, read> B: array<f16>;
@group(0) @binding(2) var<storage, read_write> C: array<f16>; @group(0) @binding(2) var<storage, read_write> C: array<f16>;
@group(0) @binding(3) var<uniform> size: u32;
struct TensorInfo {
size: u32,
};
@group(0) @binding(3) var<uniform> info: TensorInfo;
@compute @workgroup_size(256) @compute @workgroup_size(256)
fn main( fn main(
@ -21,7 +17,7 @@ fn main(
var index = global_id.x; var index = global_id.x;
// 3. Stride through the tensor elements // 3. Stride through the tensor elements
while (index < info.size) { while (index < size) {
C[index] = A[index] + B[index]; C[index] = A[index] + B[index];
index += total_threads; // Jump forward by the total thread count index += total_threads; // Jump forward by the total thread count
} }