An example using compute shader?

[elgs]

I am new here. Can you show an simple example using TypeGPU of a compute shader that will accept an array and increase each element by 1 in parallel on gpu? Thanks.

[iwoplaza]

Hi! Here's an example of such compute shader:

import tgpu from 'typegpu';
import * as d from 'typegpu/data';

// Allocating an array in VRAM
const resultMutable = root.createMutable(d.arrayOf(d.f32, 100), /* optional initial value */);

// You can always override the initial value at any point in the program
// resultMutable.write([ ... ]);

const pipeline = root['~unstable'].createGuardedComputePipeline((i) => {
  'use gpu';
  // Accessing the data directly on the GPU via `.$`
  resultMutable.$[i] += 1;
});

// Dispatching parallel work, one thread per array element
pipeline.dispatchThreads(100);

// You can download the result, or dispatch more shaders that work on the GPU memory that's
// already updated.
const result = await resultMutable.read();

Let me know if there's something you'd like me to clarify or explain further.