dom/webgpu/SupportedLimits.cpp

   1 /* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
   2 /* This Source Code Form is subject to the terms of the Mozilla Public
   3  * License, v. 2.0. If a copy of the MPL was not distributed with this
   4  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
   5
   6 #include "SupportedLimits.h"
   7 #include "Adapter.h"
   8 #include "mozilla/dom/WebGPUBinding.h"
   9
  10 namespace mozilla::webgpu {
  11
  12 GPU_IMPL_CYCLE_COLLECTION(SupportedLimits, mParent)
  13 GPU_IMPL_JS_WRAP(SupportedLimits)
  14
  15 SupportedLimits::SupportedLimits(Adapter* const aParent,
  16                                  const ffi::WGPULimits& aLimits)
  17     : ChildOf(aParent), mFfi(std::make_unique<ffi::WGPULimits>(aLimits)) {}
  18
  19 SupportedLimits::~SupportedLimits() = default;
  20
  21 uint64_t GetLimit(const ffi::WGPULimits& limits, const Limit limit) {
  22   switch (limit) {
  23     case Limit::MaxTextureDimension1D:
  24       return limits.max_texture_dimension_1d;
  25     case Limit::MaxTextureDimension2D:
  26       return limits.max_texture_dimension_2d;
  27     case Limit::MaxTextureDimension3D:
  28       return limits.max_texture_dimension_3d;
  29     case Limit::MaxTextureArrayLayers:
  30       return limits.max_texture_array_layers;
  31     case Limit::MaxBindGroups:
  32       return limits.max_bind_groups;
  33     case Limit::MaxBindGroupsPlusVertexBuffers:
  34       // Not in ffi::WGPULimits, so synthesize:
  35       return GetLimit(limits, Limit::MaxBindGroups) +
  36              GetLimit(limits, Limit::MaxVertexBuffers);
  37     case Limit::MaxBindingsPerBindGroup:
  38       return limits.max_bindings_per_bind_group;
  39     case Limit::MaxDynamicUniformBuffersPerPipelineLayout:
  40       return limits.max_dynamic_uniform_buffers_per_pipeline_layout;
  41     case Limit::MaxDynamicStorageBuffersPerPipelineLayout:
  42       return limits.max_dynamic_storage_buffers_per_pipeline_layout;
  43     case Limit::MaxSampledTexturesPerShaderStage:
  44       return limits.max_sampled_textures_per_shader_stage;
  45     case Limit::MaxSamplersPerShaderStage:
  46       return limits.max_samplers_per_shader_stage;
  47     case Limit::MaxStorageBuffersPerShaderStage:
  48       return limits.max_storage_buffers_per_shader_stage;
  49     case Limit::MaxStorageTexturesPerShaderStage:
  50       return limits.max_storage_textures_per_shader_stage;
  51     case Limit::MaxUniformBuffersPerShaderStage:
  52       return limits.max_uniform_buffers_per_shader_stage;
  53     case Limit::MaxUniformBufferBindingSize:
  54       return limits.max_uniform_buffer_binding_size;
  55     case Limit::MaxStorageBufferBindingSize:
  56       return limits.max_storage_buffer_binding_size;
  57     case Limit::MinUniformBufferOffsetAlignment:
  58       return limits.min_uniform_buffer_offset_alignment;
  59     case Limit::MinStorageBufferOffsetAlignment:
  60       return limits.min_storage_buffer_offset_alignment;
  61     case Limit::MaxVertexBuffers:
  62       return limits.max_vertex_buffers;
  63     case Limit::MaxBufferSize:
  64       return limits.max_buffer_size;
  65     case Limit::MaxVertexAttributes:
  66       return limits.max_vertex_attributes;
  67     case Limit::MaxVertexBufferArrayStride:
  68       return limits.max_vertex_buffer_array_stride;
  69     case Limit::MaxInterStageShaderComponents:
  70       return limits.max_inter_stage_shader_components;
  71     case Limit::MaxInterStageShaderVariables:
  72       return 16;  // From the spec. (not in ffi::WGPULimits)
  73     case Limit::MaxColorAttachments:
  74       return 8;  // From the spec. (not in ffi::WGPULimits)
  75     case Limit::MaxColorAttachmentBytesPerSample:
  76       return 32;  // From the spec. (not in ffi::WGPULimits)
  77     case Limit::MaxComputeWorkgroupStorageSize:
  78       return limits.max_compute_workgroup_storage_size;
  79     case Limit::MaxComputeInvocationsPerWorkgroup:
  80       return limits.max_compute_invocations_per_workgroup;
  81     case Limit::MaxComputeWorkgroupSizeX:
  82       return limits.max_compute_workgroup_size_x;
  83     case Limit::MaxComputeWorkgroupSizeY:
  84       return limits.max_compute_workgroup_size_y;
  85     case Limit::MaxComputeWorkgroupSizeZ:
  86       return limits.max_compute_workgroup_size_z;
  87     case Limit::MaxComputeWorkgroupsPerDimension:
  88       return limits.max_compute_workgroups_per_dimension;
  89   }
  90   MOZ_CRASH("Bad Limit");
  91 }
  92
  93 void SetLimit(ffi::WGPULimits* const limits, const Limit limit,
  94               const double val) {
  95   const auto autoVal = LazyAssertedCast(static_cast<uint64_t>(val));
  96   switch (limit) {
  97     case Limit::MaxTextureDimension1D:
  98       limits->max_texture_dimension_1d = autoVal;
  99       return;
 100     case Limit::MaxTextureDimension2D:
 101       limits->max_texture_dimension_2d = autoVal;
 102       return;
 103     case Limit::MaxTextureDimension3D:
 104       limits->max_texture_dimension_3d = autoVal;
 105       return;
 106     case Limit::MaxTextureArrayLayers:
 107       limits->max_texture_array_layers = autoVal;
 108       return;
 109     case Limit::MaxBindGroups:
 110       limits->max_bind_groups = autoVal;
 111       return;
 112     case Limit::MaxBindGroupsPlusVertexBuffers:
 113       // Not in ffi::WGPULimits, and we're allowed to give back better
 114       // limits than requested.
 115       return;
 116     case Limit::MaxBindingsPerBindGroup:
 117       limits->max_bindings_per_bind_group = autoVal;
 118       return;
 119     case Limit::MaxDynamicUniformBuffersPerPipelineLayout:
 120       limits->max_dynamic_uniform_buffers_per_pipeline_layout = autoVal;
 121       return;
 122     case Limit::MaxDynamicStorageBuffersPerPipelineLayout:
 123       limits->max_dynamic_storage_buffers_per_pipeline_layout = autoVal;
 124       return;
 125     case Limit::MaxSampledTexturesPerShaderStage:
 126       limits->max_sampled_textures_per_shader_stage = autoVal;
 127       return;
 128     case Limit::MaxSamplersPerShaderStage:
 129       limits->max_samplers_per_shader_stage = autoVal;
 130       return;
 131     case Limit::MaxStorageBuffersPerShaderStage:
 132       limits->max_storage_buffers_per_shader_stage = autoVal;
 133       return;
 134     case Limit::MaxStorageTexturesPerShaderStage:
 135       limits->max_storage_textures_per_shader_stage = autoVal;
 136       return;
 137     case Limit::MaxUniformBuffersPerShaderStage:
 138       limits->max_uniform_buffers_per_shader_stage = autoVal;
 139       return;
 140     case Limit::MaxUniformBufferBindingSize:
 141       limits->max_uniform_buffer_binding_size = autoVal;
 142       return;
 143     case Limit::MaxStorageBufferBindingSize:
 144       limits->max_storage_buffer_binding_size = autoVal;
 145       return;
 146     case Limit::MinUniformBufferOffsetAlignment:
 147       limits->min_uniform_buffer_offset_alignment = autoVal;
 148       return;
 149     case Limit::MinStorageBufferOffsetAlignment:
 150       limits->min_storage_buffer_offset_alignment = autoVal;
 151       return;
 152     case Limit::MaxVertexBuffers:
 153       limits->max_vertex_buffers = autoVal;
 154       return;
 155     case Limit::MaxBufferSize:
 156       limits->max_buffer_size = autoVal;
 157       return;
 158     case Limit::MaxVertexAttributes:
 159       limits->max_vertex_attributes = autoVal;
 160       return;
 161     case Limit::MaxVertexBufferArrayStride:
 162       limits->max_vertex_buffer_array_stride = autoVal;
 163       return;
 164     case Limit::MaxInterStageShaderComponents:
 165       limits->max_inter_stage_shader_components = autoVal;
 166       return;
 167     case Limit::MaxInterStageShaderVariables:
 168       // Not in ffi::WGPULimits, and we're allowed to give back better
 169       // limits than requested.
 170       return;
 171     case Limit::MaxColorAttachments:
 172       // Not in ffi::WGPULimits, and we're allowed to give back better
 173       // limits than requested.
 174       return;
 175     case Limit::MaxColorAttachmentBytesPerSample:
 176       // Not in ffi::WGPULimits, and we're allowed to give back better
 177       // limits than requested.
 178       return;
 179     case Limit::MaxComputeWorkgroupStorageSize:
 180       limits->max_compute_workgroup_storage_size = autoVal;
 181       return;
 182     case Limit::MaxComputeInvocationsPerWorkgroup:
 183       limits->max_compute_invocations_per_workgroup = autoVal;
 184       return;
 185     case Limit::MaxComputeWorkgroupSizeX:
 186       limits->max_compute_workgroup_size_x = autoVal;
 187       return;
 188     case Limit::MaxComputeWorkgroupSizeY:
 189       limits->max_compute_workgroup_size_y = autoVal;
 190       return;
 191     case Limit::MaxComputeWorkgroupSizeZ:
 192       limits->max_compute_workgroup_size_z = autoVal;
 193       return;
 194     case Limit::MaxComputeWorkgroupsPerDimension:
 195       limits->max_compute_workgroups_per_dimension = autoVal;
 196       return;
 197   }
 198   MOZ_CRASH("Bad Limit");
 199 }
 200
 201 }  // namespace mozilla::webgpu