GPUDevice: метод createComputePipeline()

Метод createComputePipeline() интерфейса GPUDevice создаёт GPUComputePipeline, который может управлять стадией шейдера вычислений и использоваться в GPUComputePassEncoder.

descriptor

Объект, содержащий следующие свойства:

compute

Объект, описывающий точку входа шейдера вычислений в конвейере. Этот объект может содержать следующие свойства:

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

Последовательность типов записей со структурой (id, value), представляющая значения переопределения для постоянных WGSL, которые можно переопределить в конвейере. Они ведут себя как упорядоченные карты. В каждом случае id является ключом, используемым для идентификации или выбора записи, а constant — перечисленным значением, представляющим WGSL.

В зависимости от того, какое постоянное значение вы хотите переопределить, id может представлять собой числовой идентификатор постоянного значения, если он указан, или в противном случае — идентификатор имени постоянного значения.

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

{
  // ...
  constants: {
    0: false,
    1200: 3.0,
    1300: 2.0,
    width: 20,
    depth: -1,
    height: 15,
  }
}

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

Имя функции в module , которую эта стадия будет использовать для выполнения своей работы. Соответствующая функция шейдера должна иметь атрибут @compute, чтобы быть идентифицированной как эта точка входа. Дополнительную информацию см. в Объявлении точки входа.

Вы можете опустить свойство entryPoint, если ваш код шейдера содержит единственную функцию с установленным атрибутом @compute — браузер будет использовать её в качестве точки входа по умолчанию. Если entryPoint опущено и браузер не может определить точку входа по умолчанию, генерируется GPUValidationError, а полученный GPUComputePipeline будет недействительным.

module

Объект GPUShaderModule, содержащий код WGSL, который будет выполняться этой программируемой стадией.

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

Строка, предоставляющая метку, которая может использоваться для идентификации объекта, например, в сообщениях GPUError или предупреждениях консоли.

layout

Определяет макет (структуру, назначение и тип) всех ресурсов GPU (буферы, текстуры и т. д.), используемых во время выполнения конвейера. Возможные значения:

• Объект GPUPipelineLayout, созданный с помощью GPUDevice.createPipelineLayout(), который позволяет GPU определить наиболее эффективный способ запуска конвейера заранее.
• Строка "auto", которая заставляет конвейер генерировать неявный макет группы связей на основе любых связей, определённых в коде шейдера. Если используется "auto", сгенерированные макеты групп связей могут быть использованы только с текущим конвейером.

Экземпляр объекта GPUComputePipeline.

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

• Размер хранилища рабочей группы, используемый module , который указан в свойстве compute, меньше или равен пределу maxComputeWorkgroupStorageSize предела устройства GPUDevice.
• module использует меньше или равное количеству вызовов вычислений на рабочую группу, чем предел maxComputeInvocationsPerWorkgroup предела устройства GPUDevice.
• Размер рабочей группы module меньше или равен соответствующему пределу maxComputeWorkgroupSizeX, maxComputeWorkgroupSizeY, или maxComputeWorkgroupSizeZ предела устройства GPUDevice.
• Если свойство entryPoint опущено, код шейдера содержит единственную функцию точки входа шейдера вычислений, которую браузер может использовать в качестве точки входа по умолчанию.

Наш базовый пример вычислений демонстрирует процесс:

• Создание макета группы связей с помощью GPUDevice.createBindGroupLayout().
• Использование bindGroupLayout в GPUDevice.createPipelineLayout() для создания GPUPipelineLayout.
• Непосредственное использование этого значения в вызове createComputePipeline() для создания GPUComputePipeline.

// ...

const bindGroupLayout = device.createBindGroupLayout({
  entries: [
    {
      binding: 0,
      visibility: GPUShaderStage.COMPUTE,
      buffer: {
        type: "storage",
      },
    },
  ],
});

const computePipeline = device.createComputePipeline({
  layout: device.createPipelineLayout({
    bindGroupLayouts: [bindGroupLayout],
  }),
  compute: {
    module: shaderModule,
    entryPoint: "main",
  },
});

// ...

• API WebGPU