Enable subgroup cache

jiangzhaoming · Nov 7, 2024 · 08d98ee · 08d98ee
1 parent 5c5495d
commit 08d98ee
Show file tree

Hide file tree

Showing 3 changed files with 669 additions and 228 deletions.
diff --git a/js/web/lib/wasm/jsep/backend-webgpu.ts b/js/web/lib/wasm/jsep/backend-webgpu.ts
@@ -251,6 +251,9 @@ export class WebGpuBackend {
     if (adapter.features.has('shader-f16')) {
       requiredFeatures.push('shader-f16');
     }
+    if (adapter.features.has('chromium-experimental-subgroups')) {
+      requiredFeatures.push('chromium-experimental-subgroups' as GPUFeatureName);
+    }
 
     this.device = await adapter.requestDevice(deviceDescriptor);
     this.adapterInfo = new AdapterInfoImpl(adapter.info || (await adapter.requestAdapterInfo()));