GPUComputePassEncoder: метод dispatchWorkgroups()

Метод dispatchWorkgroups() интерфейса GPUComputePassEncoder отправляет определённую сетку рабочих групп для выполнения работы, выполняемой текущим GPUComputePipeline (т.е. установленной с помощью GPUComputePassEncoder.setPipeline()).

workgroupCountX

Размер X сетки рабочих групп для отправки.

workgroupCountY Необязательно

Размер Y сетки рабочих групп для отправки. Если опущен, по умолчанию workgroupCountY равен 1.

workgroupCountZ Необязательно

Размер Z сетки рабочих групп для отправки. Если опущен, по умолчанию workgroupCountZ равен 1.

None (Undefined).

При вызове dispatchWorkgroups() должны быть соблюдены следующие критерии; в противном случае генерируется GPUValidationError, и GPUComputePassEncoder становится недействительным:

• workgroupCountX, workgroupCountY, и workgroupCountZ все меньше или равны пределу maxComputeWorkgroupsPerDimension устройства GPUDevice лимита.

В нашем демо вычислений несколько команд записываются через GPUCommandEncoder. Большинство этих команд исходят от GPUComputePassEncoder, созданного с помощью beginComputePass().

В начале кода мы устанавливаем глобальный размер буфера в 1000. Также обратите внимание, что размер рабочей группы в шейдере установлен на 64.

const BUFFER_SIZE = 1000;

// Compute shader
const shader = `
@group(0) @binding(0)
var<storage, read_write> output: array<f32>;

@compute @workgroup_size(64)

...

`;

Позже в коде параметр dispatchWorkgroups() workgroupCountX устанавливается на основе глобального размера буфера и количества рабочих групп шейдера.

// ...

// Create GPUCommandEncoder to encode commands to issue to the GPU
const commandEncoder = device.createCommandEncoder();

// Initiate render pass
const passEncoder = commandEncoder.beginComputePass();

// Issue commands
passEncoder.setPipeline(computePipeline);
passEncoder.setBindGroup(0, bindGroup);
passEncoder.dispatchWorkgroups(Math.ceil(BUFFER_SIZE / 64));

// End the render pass
passEncoder.end();

// Copy output buffer to staging buffer
commandEncoder.copyBufferToBuffer(
  output,
  0, // Source offset
  stagingBuffer,
  0, // Destination offset
  BUFFER_SIZE,
);

// End frame by passing array of command buffers to command queue for execution
device.queue.submit([commandEncoder.finish()]);

// ...

• API WebGPU