oidn-webgpu 0.0.0-stage → 0.6.0

package/lib/WGPUFullQuadPass.js

@@ -0,0 +1,261 @@
+export const fullScreenQuadVertexShaderWGSL = /*wgsl */ `
+@vertex
+fn main(
+  @builtin(vertex_index) vertexIndex: u32
+) -> @builtin(position) vec4f {
+  const pos = array(
+    vec2(-1.0, -1.0), vec2(1.0, -1.0), vec2(-1.0, 1.0),
+    vec2(-1.0, 1.0), vec2(1.0, -1.0), vec2(1.0, 1.0),
+  );
+
+  return vec4<f32>(pos[vertexIndex], 0.0, 1.0);
+}
+`;
+function isTextureParamsEqual(params, other) {
+    return params.format === other.format;
+}
+export class WGPUFullQuadPass {
+    _label;
+    _device;
+    _outputTextures = {};
+    _pipeline;
+    _bindGroups = [];
+    _needsUpdatePipeline = true;
+    /**
+     * When render to canvas
+     */
+    _renderToScreen;
+    _inputs = [];
+    _outputs = [];
+    _uniforms = [];
+    _uniformBuffers = {};
+    _width = 10;
+    _height = 10;
+    _fsCode = '';
+    _fsMain;
+    _fsDefine;
+    constructor(label, device, opts) {
+        this._label = label;
+        this._device = device;
+        this._fsMain = opts.fsMain;
+        this._fsDefine = opts.fsDefine;
+        this._inputs = opts.inputs;
+        this._outputs = opts.outputs;
+        this._uniforms = opts.uniforms;
+        opts.uniforms.forEach((uniform) => {
+            this._uniformBuffers[uniform.label] = device.createBuffer({
+                label: this._label,
+                size: uniform.data.byteLength,
+                usage: GPUBufferUsage.UNIFORM | GPUBufferUsage.COPY_DST
+            });
+            this._device.queue.writeBuffer(this._uniformBuffers[uniform.label], 0, uniform.data);
+        });
+    }
+    setSize(width, height) {
+        width = Math.ceil(width);
+        height = Math.ceil(height);
+        const sizeChanged = width !== this._width || height !== this._height;
+        this._width = width;
+        this._height = height;
+        if (sizeChanged) {
+            this._resizeOutputTextures();
+            this._needsUpdatePipeline = true;
+        }
+    }
+    setOutputParams(outputParams) {
+        this._renderToScreen = undefined;
+        this._updateOutputTextures(outputParams);
+        this._needsUpdatePipeline = true;
+    }
+    setRenderToScreen(screenTexture, presentationFormat) {
+        this._renderToScreen = {
+            screenTexture,
+            presentationFormat
+        };
+    }
+    setUniform(label, data) {
+        const buffer = this._uniformBuffers[label];
+        this._device.queue.writeBuffer(buffer, 0, data);
+    }
+    getOutputTexture(name) {
+        return this._outputTextures[name].texture;
+    }
+    dispose() {
+        Object.keys(this._uniformBuffers).forEach((key) => {
+            this._uniformBuffers[key].destroy();
+        });
+        Object.keys(this._outputTextures).forEach((key) => {
+            this._outputTextures[key].texture.destroy();
+        });
+    }
+    _createTexture(params) {
+        return this._device.createTexture({
+            label: this._label,
+            size: {
+                width: this._width,
+                height: this._height,
+                depthOrArrayLayers: 1
+            },
+            format: params.format,
+            usage: GPUTextureUsage.RENDER_ATTACHMENT | GPUTextureUsage.TEXTURE_BINDING
+        });
+    }
+    _resizeOutputTextures() {
+        const outputTextures = this._outputTextures;
+        for (const key in outputTextures) {
+            const { texture, params } = outputTextures[key];
+            texture.destroy();
+            outputTextures[key].texture = this._createTexture(params);
+        }
+    }
+    _updateOutputTextures(outputParams) {
+        const outputTextures = this._outputTextures;
+        for (const key in outputParams) {
+            const params = outputParams[key];
+            if (!isTextureParamsEqual(params, outputTextures[key]?.params || {})) {
+                outputTextures[key]?.texture.destroy();
+                const texture = this._createTexture(params);
+                outputTextures[key] = {
+                    texture,
+                    params
+                };
+            }
+        }
+    }
+    _updatePipeline() {
+        if (!this._needsUpdatePipeline) {
+            return;
+        }
+        this._needsUpdatePipeline = false;
+        const device = this._device;
+        const fsCode = this._getFullFs();
+        if (fsCode === this._fsCode) {
+            return;
+        }
+        const { screenTexture, presentationFormat } = this._renderToScreen || {};
+        this._fsCode = fsCode;
+        this._pipeline = device.createRenderPipeline({
+            label: this._label,
+            layout: 'auto',
+            vertex: {
+                module: device.createShaderModule({
+                    label: this._label,
+                    code: fullScreenQuadVertexShaderWGSL
+                }),
+                entryPoint: 'main'
+            },
+            fragment: {
+                module: device.createShaderModule({
+                    label: this._label,
+                    code: fsCode
+                }),
+                entryPoint: 'main',
+                targets: screenTexture
+                    ? [
+                        {
+                            format: presentationFormat
+                        }
+                    ]
+                    : this._outputs.map((key) => ({
+                        format: this._outputTextures[key].params.format
+                    }))
+            },
+            primitive: {
+                topology: 'triangle-list'
+            }
+        });
+        this._updateBindGroups();
+    }
+    _getFullFs() {
+        const inputs = this._inputs;
+        const hasInputs = inputs.length > 0;
+        const fs = `
+${inputs
+            .sort()
+            .map((textureName, idx) => `@group(0) @binding(${idx}) var ${textureName}: texture_2d<f32>;`)
+            .join('\n')}
+${this._uniforms
+            .map((uniform, idx) => `@group(${hasInputs ? 1 : 0}) @binding(${idx}) var<uniform> ${uniform.label}: ${uniform.type};`)
+            .join('\n')}
+
+struct FSOutput {
+${this._outputs
+            .map((name, idx) => `@location(${idx}) ${name}: vec4f,`)
+            .join('\n')}
+}
+${this._fsDefine ?? ''}
+@fragment
+fn main(
+  @builtin(position) coord: vec4f
+) -> FSOutput {
+  var uv = vec2i(floor(coord.xy));
+  var output: FSOutput;
+${this._fsMain}
+  return output;
+}
+`;
+        return fs;
+    }
+    _updateBindGroups() {
+        const bindGroups = [];
+        const device = this._device;
+        //TODO
+        const uniformBindGroupIndex = this._inputs.length > 0 ? 1 : 0;
+        if (this._uniforms.length > 0) {
+            bindGroups[uniformBindGroupIndex] = device.createBindGroup({
+                label: this._label,
+                layout: this._pipeline.getBindGroupLayout(uniformBindGroupIndex),
+                entries: this._uniforms.map((uniform, idx) => ({
+                    binding: idx,
+                    resource: {
+                        buffer: this._uniformBuffers[uniform.label]
+                    }
+                }))
+            });
+        }
+        this._bindGroups = bindGroups;
+    }
+    createPass(commandEncoder, inputTextures) {
+        this._updatePipeline();
+        // TODO createBindGrou every time?
+        if (this._inputs.length > 0) {
+            this._bindGroups[0] = this._device.createBindGroup({
+                label: this._label,
+                layout: this._pipeline.getBindGroupLayout(0),
+                entries: this._inputs.map((textureName, idx) => ({
+                    binding: idx,
+                    // TODO
+                    resource: inputTextures[textureName].createView()
+                }))
+            });
+        }
+        // Begin the render pass
+        const renderPass = commandEncoder.beginRenderPass({
+            colorAttachments: this._renderToScreen
+                ? [
+                    {
+                        view: this._renderToScreen.screenTexture.createView(),
+                        clearValue: { r: 0, g: 0, b: 0, a: 0 },
+                        storeOp: 'store',
+                        loadOp: 'clear'
+                    }
+                ]
+                : this._outputs.map((textureName) => ({
+                    view: this._outputTextures[textureName].texture.createView(),
+                    clearValue: { r: 0, g: 0, b: 0, a: 0 },
+                    loadOp: 'clear',
+                    storeOp: 'store'
+                }))
+        });
+        // Draw a full quad
+        renderPass.setPipeline(this._pipeline);
+        // Bind groups
+        this._bindGroups.forEach((bindGroup, idx) => {
+            renderPass.setBindGroup(idx, bindGroup);
+        });
+        renderPass.draw(6, 1, 0, 0);
+        // End the render pass
+        renderPass.end();
+    }
+}
+//# sourceMappingURL=WGPUFullQuadPass.js.map

package/lib/WGPUFullQuadPass.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"WGPUFullQuadPass.js","sourceRoot":"","sources":["../src/WGPUFullQuadPass.ts"],"names":[],"mappings":"AAAA,MAAM,CAAC,MAAM,8BAA8B,GAAG,SAAS,CAAC;;;;;;;;;;;;CAYvD,CAAC;AAEF,SAAS,oBAAoB,CAC3B,MAA8B,EAC9B,KAA6B;IAE7B,OAAO,MAAM,CAAC,MAAM,KAAK,KAAK,CAAC,MAAM,CAAC;AACxC,CAAC;AAWD,MAAM,OAAO,gBAAgB;IACnB,MAAM,CAAC;IAEP,OAAO,CAAC;IACR,eAAe,GAMnB,EAAE,CAAC;IAEC,SAAS,CAAqB;IAC9B,WAAW,GAAmB,EAAE,CAAC;IACjC,oBAAoB,GAAG,IAAI,CAAC;IACpC;;OAEG;IACK,eAAe,CAGrB;IACM,OAAO,GAAa,EAAE,CAAC;IACvB,QAAQ,GAAa,EAAE,CAAC;IACxB,SAAS,GAAc,EAAE,CAAC;IAC1B,eAAe,GAA8B,EAAE,CAAC;IAEhD,MAAM,GAAG,EAAE,CAAC;IACZ,OAAO,GAAG,EAAE,CAAC;IAEb,OAAO,GAAG,EAAE,CAAC;IACb,OAAO,CAAC;IACR,SAAS,CAAC;IAElB,YACE,KAAa,EACb,MAAiB,EACjB,IAMC;QAED,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC;QACpB,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;QACtB,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC;QAC3B,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,QAAQ,CAAC;QAC/B,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC;QAC3B,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC;QAC7B,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,QAAQ,CAAC;QAE/B,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;YAChC,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG,MAAM,CAAC,YAAY,CAAC;gBACxD,KAAK,EAAE,IAAI,CAAC,MAAM;gBAClB,IAAI,EAAE,OAAO,CAAC,IAAI,CAAC,UAAU;gBAC7B,KAAK,EAAE,cAAc,CAAC,OAAO,GAAG,cAAc,CAAC,QAAQ;aACxD,CAAC,CAAC;YACH,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,WAAW,CAC5B,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,KAAK,CAAC,EACnC,CAAC,EACD,OAAO,CAAC,IAAI,CACb,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC;IAED,OAAO,CAAC,KAAa,EAAE,MAAc;QACnC,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACzB,MAAM,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC3B,MAAM,WAAW,GAAG,KAAK,KAAK,IAAI,CAAC,MAAM,IAAI,MAAM,KAAK,IAAI,CAAC,OAAO,CAAC;QACrE,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC;QACpB,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;QACtB,IAAI,WAAW,EAAE,CAAC;YAChB,IAAI,CAAC,qBAAqB,EAAE,CAAC;YAC7B,IAAI,CAAC,oBAAoB,GAAG,IAAI,CAAC;QACnC,CAAC;IACH,CAAC;IAED,eAAe,CAAC,YAA+C;QAC7D,IAAI,CAAC,eAAe,GAAG,SAAS,CAAC;QACjC,IAAI,CAAC,qBAAqB,CAAC,YAAY,CAAC,CAAC;QACzC,IAAI,CAAC,oBAAoB,GAAG,IAAI,CAAC;IACnC,CAAC;IAED,iBAAiB,CACf,aAAyB,EACzB,kBAAoC;QAEpC,IAAI,CAAC,eAAe,GAAG;YACrB,aAAa;YACb,kBAAkB;SACnB,CAAC;IACJ,CAAC;IAED,UAAU,CAAC,KAAa,EAAE,IAA6C;QACrE,MAAM,MAAM,GAAG,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;QAC3C,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,EAAE,IAAI,CAAC,CAAC;IAClD,CAAC;IAED,gBAAgB,CAAC,IAAO;QACtB,OAAO,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC;IAC5C,CAAC;IAED,OAAO;QACL,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE;YAC/C,IAAI,CAAC,eAAuB,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,CAAC;QAC/C,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE;YAC/C,IAAI,CAAC,eAAuB,CAAC,GAAG,CAAC,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;QACvD,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,cAAc,CAAC,MAA8B;QACnD,OAAO,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC;YAChC,KAAK,EAAE,IAAI,CAAC,MAAM;YAClB,IAAI,EAAE;gBACJ,KAAK,EAAE,IAAI,CAAC,MAAM;gBAClB,MAAM,EAAE,IAAI,CAAC,OAAO;gBACpB,kBAAkB,EAAE,CAAC;aACtB;YACD,MAAM,EAAE,MAAM,CAAC,MAAM;YACrB,KAAK,EAAE,eAAe,CAAC,iBAAiB,GAAG,eAAe,CAAC,eAAe;SAC3E,CAAC,CAAC;IACL,CAAC;IAEO,qBAAqB;QAC3B,MAAM,cAAc,GAAG,IAAI,CAAC,eAAe,CAAC;QAC5C,KAAK,MAAM,GAAG,IAAI,cAAc,EAAE,CAAC;YACjC,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,GAAG,cAAc,CAAC,GAAG,CAAC,CAAC;YAChD,OAAO,CAAC,OAAO,EAAE,CAAC;YAClB,cAAc,CAAC,GAAG,CAAC,CAAC,OAAO,GAAG,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;QAC5D,CAAC;IACH,CAAC;IAEO,qBAAqB,CAC3B,YAAoD;QAEpD,MAAM,cAAc,GAAG,IAAI,CAAC,eAAe,CAAC;QAC5C,KAAK,MAAM,GAAG,IAAI,YAAY,EAAE,CAAC;YAC/B,MAAM,MAAM,GAAG,YAAY,CAAC,GAAG,CAAC,CAAC;YACjC,IACE,CAAC,oBAAoB,CACnB,MAAM,EACN,cAAc,CAAC,GAAG,CAAC,EAAE,MAAM,IAAK,EAA6B,CAC9D,EACD,CAAC;gBACD,cAAc,CAAC,GAAG,CAAC,EAAE,OAAO,CAAC,OAAO,EAAE,CAAC;gBACvC,MAAM,OAAO,GAAG,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;gBAC5C,cAAc,CAAC,GAAG,CAAC,GAAG;oBACpB,OAAO;oBACP,MAAM;iBACP,CAAC;YACJ,CAAC;QACH,CAAC;IACH,CAAC;IAEO,eAAe;QACrB,IAAI,CAAC,IAAI,CAAC,oBAAoB,EAAE,CAAC;YAC/B,OAAO;QACT,CAAC;QACD,IAAI,CAAC,oBAAoB,GAAG,KAAK,CAAC;QAClC,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC;QAC5B,MAAM,MAAM,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;QACjC,IAAI,MAAM,KAAK,IAAI,CAAC,OAAO,EAAE,CAAC;YAC5B,OAAO;QACT,CAAC;QACD,MAAM,EAAE,aAAa,EAAE,kBAAkB,EAAE,GAAG,IAAI,CAAC,eAAe,IAAI,EAAE,CAAC;QACzE,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;QACtB,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,oBAAoB,CAAC;YAC3C,KAAK,EAAE,IAAI,CAAC,MAAM;YAClB,MAAM,EAAE,MAAM;YACd,MAAM,EAAE;gBACN,MAAM,EAAE,MAAM,CAAC,kBAAkB,CAAC;oBAChC,KAAK,EAAE,IAAI,CAAC,MAAM;oBAClB,IAAI,EAAE,8BAA8B;iBACrC,CAAC;gBACF,UAAU,EAAE,MAAM;aACnB;YACD,QAAQ,EAAE;gBACR,MAAM,EAAE,MAAM,CAAC,kBAAkB,CAAC;oBAChC,KAAK,EAAE,IAAI,CAAC,MAAM;oBAClB,IAAI,EAAE,MAAM;iBACb,CAAC;gBACF,UAAU,EAAE,MAAM;gBAClB,OAAO,EAAE,aAAa;oBACpB,CAAC,CAAC;wBACE;4BACE,MAAM,EAAE,kBAAmB;yBAC5B;qBACF;oBACH,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;wBAC1B,MAAM,EAAE,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC,MAAM,CAAC,MAAM;qBAChD,CAAC,CAAC;aACR;YACD,SAAS,EAAE;gBACT,QAAQ,EAAE,eAAe;aAC1B;SACF,CAAC,CAAC;QACH,IAAI,CAAC,iBAAiB,EAAE,CAAC;IAC3B,CAAC;IAEO,UAAU;QAChB,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC;QAC5B,MAAM,SAAS,GAAG,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC;QACpC,MAAM,EAAE,GAAG;EACb,MAAM;aACL,IAAI,EAAE;aACN,GAAG,CACF,CAAC,WAAW,EAAE,GAAG,EAAE,EAAE,CACnB,sBAAsB,GAAG,SAAS,WAAW,oBAAoB,CACpE;aACA,IAAI,CAAC,IAAI,CAAC;EACX,IAAI,CAAC,SAAS;aACb,GAAG,CACF,CAAC,OAAO,EAAE,GAAG,EAAE,EAAE,CACf,UAAU,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,cAAc,GAAG,kBAC1C,OAAO,CAAC,KACV,KAAK,OAAO,CAAC,IAAI,GAAG,CACvB;aACA,IAAI,CAAC,IAAI,CAAC;;;EAGX,IAAI,CAAC,QAAQ;aACZ,GAAG,CAAC,CAAC,IAAI,EAAE,GAAG,EAAE,EAAE,CAAC,aAAa,GAAG,KAAK,IAAI,UAAU,CAAC;aACvD,IAAI,CAAC,IAAI,CAAC;;EAEX,IAAI,CAAC,SAAS,IAAI,EAAE;;;;;;;EAOpB,IAAI,CAAC,OAAO;;;CAGb,CAAC;QAEE,OAAO,EAAE,CAAC;IACZ,CAAC;IAEO,iBAAiB;QACvB,MAAM,UAAU,GAAmB,EAAE,CAAC;QACtC,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC;QAE5B,MAAM;QACN,MAAM,qBAAqB,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAC9D,IAAI,IAAI,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC9B,UAAU,CAAC,qBAAqB,CAAC,GAAG,MAAM,CAAC,eAAe,CAAC;gBACzD,KAAK,EAAE,IAAI,CAAC,MAAM;gBAClB,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,kBAAkB,CAAC,qBAAqB,CAAC;gBAChE,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,GAAG,CACzB,CAAC,OAAO,EAAE,GAAG,EAAE,EAAE,CACf,CAAC;oBACC,OAAO,EAAE,GAAG;oBACZ,QAAQ,EAAE;wBACR,MAAM,EAAE,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,KAAK,CAAC;qBAC5C;iBAC2B,CAAA,CACjC;aACF,CAAC,CAAC;QACL,CAAC;QAED,IAAI,CAAC,WAAW,GAAG,UAAU,CAAC;IAChC,CAAC;IAED,UAAU,CACR,cAAiC,EACjC,aAAoC;QAEpC,IAAI,CAAC,eAAe,EAAE,CAAC;QAEvB,kCAAkC;QAClC,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC5B,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC;gBACjD,KAAK,EAAE,IAAI,CAAC,MAAM;gBAClB,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,kBAAkB,CAAC,CAAC,CAAC;gBAC5C,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;oBAC/C,OAAO,EAAE,GAAG;oBACZ,OAAO;oBACP,QAAQ,EAAE,aAAa,CAAC,WAAgB,CAAC,CAAC,UAAU,EAAE;iBACvD,CAAC,CAAC;aACJ,CAAC,CAAC;QACL,CAAC;QACD,wBAAwB;QACxB,MAAM,UAAU,GAAG,cAAc,CAAC,eAAe,CAAC;YAChD,gBAAgB,EAAE,IAAI,CAAC,eAAe;gBACpC,CAAC,CAAC;oBACE;wBACE,IAAI,EAAE,IAAI,CAAC,eAAe,CAAC,aAAa,CAAC,UAAU,EAAE;wBACrD,UAAU,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE;wBACtC,OAAO,EAAE,OAAqB;wBAC9B,MAAM,EAAE,OAAoB;qBAC7B;iBACF;gBACH,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CACf,CAAC,WAAW,EAAE,EAAE,CACd,CAAC;oBACC,IAAI,EAAE,IAAI,CAAC,eAAe,CAAC,WAAW,CAAC,CAAC,OAAO,CAAC,UAAU,EAAE;oBAC5D,UAAU,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE;oBACtC,MAAM,EAAE,OAAoB;oBAC5B,OAAO,EAAE,OAAqB;iBACS,CAAA,CAC5C;SACN,CAAC,CAAC;QAEH,mBAAmB;QACnB,UAAU,CAAC,WAAW,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACvC,cAAc;QACd,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,GAAG,EAAE,EAAE;YAC1C,UAAU,CAAC,YAAY,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;QACH,UAAU,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;QAC5B,sBAAsB;QACtB,UAAU,CAAC,GAAG,EAAE,CAAC;IACnB,CAAC;CACF"}

package/lib/backend.d.ts

@@ -0,0 +1 @@
+export declare function initWebGPUBackend(): Promise<GPUDevice>;

package/lib/backend.js

@@ -0,0 +1,39 @@
+export async function initWebGPUBackend() {
+    if (!navigator.gpu)
+        throw new Error('WebGPU is not available');
+    const gpuDescriptor = {
+        powerPreference: 'high-performance'
+    };
+    const adapter = await navigator.gpu.requestAdapter(gpuDescriptor);
+    if (!adapter)
+        throw new Error('No WebGPU adapter is available');
+    const deviceDescriptor = {};
+    const requiredFeatures = [];
+    if (adapter.features.has('timestamp-query')) {
+        requiredFeatures.push('timestamp-query');
+    }
+    if (adapter.features.has('bgra8unorm-storage')) {
+        requiredFeatures.push('bgra8unorm-storage');
+    }
+    if (adapter.features.has('shader-f16')) {
+        requiredFeatures.push('shader-f16');
+    }
+    // Experimental: only exposed by Chrome with unsafe WebGPU features enabled.
+    for (const feature of ['subgroups', 'chromium-experimental-subgroup-matrix']) {
+        if (adapter.features.has(feature)) {
+            requiredFeatures.push(feature);
+        }
+    }
+    deviceDescriptor.requiredFeatures = requiredFeatures;
+    const adapterLimits = adapter.limits;
+    deviceDescriptor.requiredLimits = {
+        maxComputeWorkgroupStorageSize: adapterLimits.maxComputeWorkgroupStorageSize,
+        maxComputeWorkgroupsPerDimension: adapterLimits.maxComputeWorkgroupsPerDimension,
+        maxStorageBufferBindingSize: adapterLimits.maxStorageBufferBindingSize,
+        maxBufferSize: adapterLimits.maxBufferSize,
+        maxComputeWorkgroupSizeX: adapterLimits.maxComputeWorkgroupSizeX,
+        maxComputeInvocationsPerWorkgroup: adapterLimits.maxComputeInvocationsPerWorkgroup
+    };
+    return adapter.requestDevice(deviceDescriptor);
+}
+//# sourceMappingURL=backend.js.map

package/lib/backend.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"backend.js","sourceRoot":"","sources":["../src/backend.ts"],"names":[],"mappings":"AAAA,MAAM,CAAC,KAAK,UAAU,iBAAiB;IACrC,IAAI,CAAC,SAAS,CAAC,GAAG;QAAE,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;IAC/D,MAAM,aAAa,GAA6B;QAC9C,eAAe,EAAE,kBAAkB;KACpC,CAAC;IAEF,MAAM,OAAO,GAAG,MAAM,SAAS,CAAC,GAAG,CAAC,cAAc,CAAC,aAAa,CAAC,CAAC;IAClE,IAAI,CAAC,OAAO;QAAE,MAAM,IAAI,KAAK,CAAC,gCAAgC,CAAC,CAAC;IAChE,MAAM,gBAAgB,GAAwB,EAAE,CAAC;IAEjD,MAAM,gBAAgB,GAAqB,EAAE,CAAC;IAC9C,IAAI,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,iBAAiB,CAAC,EAAE,CAAC;QAC5C,gBAAgB,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;IAC3C,CAAC;IACD,IAAI,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,oBAAoB,CAAC,EAAE,CAAC;QAC/C,gBAAgB,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;IAC9C,CAAC;IACD,IAAI,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,YAAY,CAAC,EAAE,CAAC;QACvC,gBAAgB,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;IACtC,CAAC;IACD,4EAA4E;IAC5E,KAAK,MAAM,OAAO,IAAI,CAAC,WAAW,EAAE,uCAAuC,CAAC,EAAE,CAAC;QAC7E,IAAI,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;YAClC,gBAAgB,CAAC,IAAI,CAAC,OAAyB,CAAC,CAAC;QACnD,CAAC;IACH,CAAC;IACD,gBAAgB,CAAC,gBAAgB,GAAG,gBAAgB,CAAC;IAErD,MAAM,aAAa,GAAG,OAAO,CAAC,MAAM,CAAC;IACrC,gBAAgB,CAAC,cAAc,GAAG;QAChC,8BAA8B,EAC5B,aAAa,CAAC,8BAA8B;QAC9C,gCAAgC,EAC9B,aAAa,CAAC,gCAAgC;QAChD,2BAA2B,EAAE,aAAa,CAAC,2BAA2B;QACtE,aAAa,EAAE,aAAa,CAAC,aAAa;QAC1C,wBAAwB,EAAE,aAAa,CAAC,wBAAwB;QAChE,iCAAiC,EAC/B,aAAa,CAAC,iCAAiC;KAClD,CAAC;IACF,OAAO,OAAO,CAAC,aAAa,CAAC,gBAAgB,CAAC,CAAC;AACjD,CAAC"}

package/lib/finalRgbShader.d.ts

@@ -0,0 +1,13 @@
+export type FinalRgbPrecision = 'fp16' | 'fp32';
+export type FinalRgbActivation = 'relu' | 'identity';
+/** Shared storage required by the 8x8 final-RGB workgroup. */
+export declare function sharedMemoryBytes(precision: FinalRgbPrecision, inputBlocks: number, cacheWeights?: boolean): number;
+/**
+ * Builds the final three-channel same-padded convolution shader.
+ *
+ * The bind-group layout and Params block intentionally match createConvShader:
+ * input, weights, bias, output, then uniform params. Weights retain the
+ * output-major packed ABI, with four vec4 values per (kernel position, input
+ * block), one for each input lane.
+ */
+export declare function createFinalRgbShader(precision: FinalRgbPrecision, activation: FinalRgbActivation, inputBlocks: number, cacheWeights?: boolean): string;

package/lib/finalRgbShader.js

@@ -0,0 +1,160 @@
+const WORKGROUP_SIZE = 8;
+const PATCH_SIZE = WORKGROUP_SIZE + 2;
+const KERNEL_ELEMENTS = 3 * 3;
+/** Shared storage required by the 8x8 final-RGB workgroup. */
+export function sharedMemoryBytes(precision, inputBlocks, cacheWeights = false) {
+    const bytesPerScalar = precision === 'fp16' ? 2 : 4;
+    const inputVec4Count = PATCH_SIZE * PATCH_SIZE * inputBlocks;
+    const weightVec4Count = cacheWeights
+        ? KERNEL_ELEMENTS * inputBlocks * 4
+        : 0;
+    const vec4Count = inputVec4Count + weightVec4Count;
+    return vec4Count * 4 * bytesPerScalar;
+}
+function storageVecType(precision) {
+    return precision === 'fp16' ? 'vec4<f16>' : 'vec4<f32>';
+}
+function shaderPreamble(precision) {
+    return precision === 'fp16' ? 'enable f16;\n' : '';
+}
+function accumulationCode(precision, inputExpression, weightExpression) {
+    const weight = (lane) => `${weightExpression}[weightBase + ${lane}u]`;
+    if (precision === 'fp16') {
+        return /* wgsl */ `
+        let inputValue = vec4<f16>(${inputExpression});
+        var partial = vec4<f16>(0.0h);
+        partial = fma(${weight(0)}, vec4<f16>(inputValue.x), partial);
+        partial = fma(${weight(1)}, vec4<f16>(inputValue.y), partial);
+        partial = fma(${weight(2)}, vec4<f16>(inputValue.z), partial);
+        partial = fma(${weight(3)}, vec4<f16>(inputValue.w), partial);
+        acc += vec4<f32>(partial);
+`;
+    }
+    return /* wgsl */ `
+        let inputValue = vec4<f32>(${inputExpression});
+        acc = fma(vec4<f32>(${weight(0)}), vec4<f32>(inputValue.x), acc);
+        acc = fma(vec4<f32>(${weight(1)}), vec4<f32>(inputValue.y), acc);
+        acc = fma(vec4<f32>(${weight(2)}), vec4<f32>(inputValue.z), acc);
+        acc = fma(vec4<f32>(${weight(3)}), vec4<f32>(inputValue.w), acc);
+`;
+}
+/**
+ * Builds the final three-channel same-padded convolution shader.
+ *
+ * The bind-group layout and Params block intentionally match createConvShader:
+ * input, weights, bias, output, then uniform params. Weights retain the
+ * output-major packed ABI, with four vec4 values per (kernel position, input
+ * block), one for each input lane.
+ */
+export function createFinalRgbShader(precision, activation, inputBlocks, cacheWeights = false) {
+    if (!Number.isInteger(inputBlocks) || inputBlocks < 1) {
+        throw new Error(`Final RGB shader requires positive input blocks, got ${inputBlocks}`);
+    }
+    const inputType = storageVecType(precision);
+    const outputType = 'vec4<f32>';
+    const inputTileValues = PATCH_SIZE * PATCH_SIZE * inputBlocks;
+    const weightTileValues = KERNEL_ELEMENTS * inputBlocks * 4;
+    const stored = activation === 'relu'
+        ? 'max(acc, vec4<f32>(0.0))'
+        : 'acc';
+    const accumulation = accumulationCode(precision, 'inputTile[patchBase + inputBlock]', cacheWeights ? 'weightTile' : 'weights');
+    return /* wgsl */ `${shaderPreamble(precision)}
+struct Params {
+  inputWidth: u32,
+  inputHeight: u32,
+  outputWidth: u32,
+  outputHeight: u32,
+  inputBlocks: u32,
+  outputBlocks: u32,
+}
+
+@group(0) @binding(0) var<storage, read> inputData: array<${inputType}>;
+@group(0) @binding(1) var<storage, read> weights: array<${inputType}>;
+@group(0) @binding(2) var<storage, read> bias: array<vec4<f32>>;
+@group(0) @binding(3) var<storage, read_write> outputData: array<${outputType}>;
+@group(0) @binding(4) var<uniform> params: Params;
+
+var<workgroup> inputTile: array<${inputType}, ${inputTileValues}>;
+${cacheWeights
+        ? `var<workgroup> weightTile: array<${inputType}, ${weightTileValues}>;`
+        : ''}
+
+@compute @workgroup_size(${WORKGROUP_SIZE}, ${WORKGROUP_SIZE}, 1)
+fn main(
+  @builtin(local_invocation_id) localId: vec3<u32>,
+  @builtin(global_invocation_id) gid: vec3<u32>,
+  @builtin(workgroup_id) workgroupId: vec3<u32>
+) {
+  let localLinear = localId.y * ${WORKGROUP_SIZE}u + localId.x;
+  for (
+    var loadIndex = localLinear;
+    loadIndex < ${inputTileValues}u;
+    loadIndex += ${WORKGROUP_SIZE * WORKGROUP_SIZE}u
+  ) {
+    let tilePixel = loadIndex / ${inputBlocks}u;
+    let inputBlock = loadIndex % ${inputBlocks}u;
+    let tileX = tilePixel % ${PATCH_SIZE}u;
+    let tileY = tilePixel / ${PATCH_SIZE}u;
+    let inputX = i32(workgroupId.x * ${WORKGROUP_SIZE}u + tileX) - 1;
+    let inputY = i32(workgroupId.y * ${WORKGROUP_SIZE}u + tileY) - 1;
+    var value = ${inputType}(0.0);
+    if (
+      inputX >= 0 && inputX < i32(params.inputWidth) &&
+      inputY >= 0 && inputY < i32(params.inputHeight)
+    ) {
+      let inputIndex =
+        (u32(inputY) * params.inputWidth + u32(inputX)) *
+        ${inputBlocks}u + inputBlock;
+      value = inputData[inputIndex];
+    }
+    inputTile[loadIndex] = value;
+  }
+
+${cacheWeights ? `  for (
+    var loadIndex = localLinear;
+    loadIndex < ${weightTileValues}u;
+    loadIndex += ${WORKGROUP_SIZE * WORKGROUP_SIZE}u
+  ) {
+    weightTile[loadIndex] = weights[loadIndex];
+  }
+
+` : ''}  // Out-of-range invocations must reach this barrier before returning.
+  workgroupBarrier();
+
+  let outputInBounds =
+    gid.x < params.outputWidth &&
+    gid.y < params.outputHeight &&
+    gid.z < params.outputBlocks;
+  if (!outputInBounds) {
+    return;
+  }
+
+  var acc = bias[gid.z];
+  for (var ky = 0u; ky < 3u; ky++) {
+    let inputY = i32(gid.y) + i32(ky) - 1;
+    if (inputY < 0 || inputY >= i32(params.inputHeight)) {
+      continue;
+    }
+    for (var kx = 0u; kx < 3u; kx++) {
+      let inputX = i32(gid.x) + i32(kx) - 1;
+      if (inputX < 0 || inputX >= i32(params.inputWidth)) {
+        continue;
+      }
+      let patchBase =
+        ((localId.y + ky) * ${PATCH_SIZE}u + localId.x + kx) *
+        ${inputBlocks}u;
+      for (var inputBlock = 0u; inputBlock < ${inputBlocks}u; inputBlock++) {
+        let weightBase =
+          ((ky * 3u + kx) * ${inputBlocks}u + inputBlock) * 4u;
+        ${accumulation}
+      }
+    }
+  }
+
+  let outputIndex =
+    (gid.y * params.outputWidth + gid.x) * ${1}u + gid.z;
+  outputData[outputIndex] = ${stored};
+}
+`;
+}
+//# sourceMappingURL=finalRgbShader.js.map

package/lib/finalRgbShader.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"finalRgbShader.js","sourceRoot":"","sources":["../src/finalRgbShader.ts"],"names":[],"mappings":"AAGA,MAAM,cAAc,GAAG,CAAC,CAAC;AACzB,MAAM,UAAU,GAAG,cAAc,GAAG,CAAC,CAAC;AACtC,MAAM,eAAe,GAAG,CAAC,GAAG,CAAC,CAAC;AAE9B,8DAA8D;AAC9D,MAAM,UAAU,iBAAiB,CAC/B,SAA4B,EAC5B,WAAmB,EACnB,YAAY,GAAG,KAAK;IAEpB,MAAM,cAAc,GAAG,SAAS,KAAK,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IACpD,MAAM,cAAc,GAAG,UAAU,GAAG,UAAU,GAAG,WAAW,CAAC;IAC7D,MAAM,eAAe,GAAG,YAAY;QAClC,CAAC,CAAC,eAAe,GAAG,WAAW,GAAG,CAAC;QACnC,CAAC,CAAC,CAAC,CAAC;IACN,MAAM,SAAS,GAAG,cAAc,GAAG,eAAe,CAAC;IACnD,OAAO,SAAS,GAAG,CAAC,GAAG,cAAc,CAAC;AACxC,CAAC;AAED,SAAS,cAAc,CAAC,SAA4B;IAClD,OAAO,SAAS,KAAK,MAAM,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,WAAW,CAAC;AAC1D,CAAC;AAED,SAAS,cAAc,CAAC,SAA4B;IAClD,OAAO,SAAS,KAAK,MAAM,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,EAAE,CAAC;AACrD,CAAC;AAED,SAAS,gBAAgB,CACvB,SAA4B,EAC5B,eAAuB,EACvB,gBAAwB;IAExB,MAAM,MAAM,GAAG,CAAC,IAAY,EAAE,EAAE,CAC9B,GAAG,gBAAgB,iBAAiB,IAAI,IAAI,CAAC;IAC/C,IAAI,SAAS,KAAK,MAAM,EAAE,CAAC;QACzB,OAAO,UAAU,CAAC;qCACe,eAAe;;wBAE5B,MAAM,CAAC,CAAC,CAAC;wBACT,MAAM,CAAC,CAAC,CAAC;wBACT,MAAM,CAAC,CAAC,CAAC;wBACT,MAAM,CAAC,CAAC,CAAC;;CAEhC,CAAC;IACA,CAAC;IACD,OAAO,UAAU,CAAC;qCACiB,eAAe;8BACtB,MAAM,CAAC,CAAC,CAAC;8BACT,MAAM,CAAC,CAAC,CAAC;8BACT,MAAM,CAAC,CAAC,CAAC;8BACT,MAAM,CAAC,CAAC,CAAC;CACtC,CAAC;AACF,CAAC;AAED;;;;;;;GAOG;AACH,MAAM,UAAU,oBAAoB,CAClC,SAA4B,EAC5B,UAA8B,EAC9B,WAAmB,EACnB,YAAY,GAAG,KAAK;IAEpB,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,WAAW,CAAC,IAAI,WAAW,GAAG,CAAC,EAAE,CAAC;QACtD,MAAM,IAAI,KAAK,CAAC,wDAAwD,WAAW,EAAE,CAAC,CAAC;IACzF,CAAC;IACD,MAAM,SAAS,GAAG,cAAc,CAAC,SAAS,CAAC,CAAC;IAC5C,MAAM,UAAU,GAAG,WAAW,CAAC;IAC/B,MAAM,eAAe,GAAG,UAAU,GAAG,UAAU,GAAG,WAAW,CAAC;IAC9D,MAAM,gBAAgB,GAAG,eAAe,GAAG,WAAW,GAAG,CAAC,CAAC;IAC3D,MAAM,MAAM,GAAG,UAAU,KAAK,MAAM;QAClC,CAAC,CAAC,0BAA0B;QAC5B,CAAC,CAAC,KAAK,CAAC;IACV,MAAM,YAAY,GAAG,gBAAgB,CACnC,SAAS,EACT,mCAAmC,EACnC,YAAY,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,SAAS,CACxC,CAAC;IAEF,OAAO,UAAU,CAAC,GAAG,cAAc,CAAC,SAAS,CAAC;;;;;;;;;;4DAUY,SAAS;0DACX,SAAS;;mEAEA,UAAU;;;kCAG3C,SAAS,KAAK,eAAe;EAC7D,YAAY;QACV,CAAC,CAAC,oCAAoC,SAAS,KAAK,gBAAgB,IAAI;QACxE,CAAC,CAAC,EAAE;;2BAEmB,cAAc,KAAK,cAAc;;;;;;kCAM1B,cAAc;;;kBAG9B,eAAe;mBACd,cAAc,GAAG,cAAc;;kCAEhB,WAAW;mCACV,WAAW;8BAChB,UAAU;8BACV,UAAU;uCACD,cAAc;uCACd,cAAc;kBACnC,SAAS;;;;;;;UAOjB,WAAW;;;;;;EAMnB,YAAY,CAAC,CAAC,CAAC;;kBAEC,gBAAgB;mBACf,cAAc,GAAG,cAAc;;;;;CAKjD,CAAC,CAAC,CAAC,EAAE;;;;;;;;;;;;;;;;;;;;;;;8BAuBwB,UAAU;UAC9B,WAAW;+CAC0B,WAAW;;8BAE5B,WAAW;UAC/B,YAAY;;;;;;6CAMuB,CAAC;8BAChB,MAAM;;CAEnC,CAAC;AACF,CAAC"}

package/lib/gpuTypes.d.ts

@@ -0,0 +1,158 @@
+/**
+ * The WebGPU subset used by the native WGSL runtime, as structural types and
+ * spec-defined flag values. A browser GPUDevice implements these operations,
+ * but its branded TypeScript types need the conversion in nativeUNet.ts;
+ * they also let an embedder supply its own device (for example one that
+ * records the work for replay in a native wgpu host) without lib.dom or
+ * WebGPU globals.
+ */
+/** GPUBufferUsage flag values from the WebGPU specification. */
+export declare const BufferUsage: {
+    readonly MAP_READ: 1;
+    readonly MAP_WRITE: 2;
+    readonly COPY_SRC: 4;
+    readonly COPY_DST: 8;
+    readonly INDEX: 16;
+    readonly VERTEX: 32;
+    readonly UNIFORM: 64;
+    readonly STORAGE: 128;
+    readonly INDIRECT: 256;
+    readonly QUERY_RESOLVE: 512;
+};
+/** GPUMapMode flag values from the WebGPU specification. */
+export declare const MapMode: {
+    readonly READ: 1;
+    readonly WRITE: 2;
+};
+export interface OIDNGPUBuffer {
+    readonly size: number;
+    readonly mapState: string;
+    getMappedRange(offset?: number, size?: number): ArrayBuffer;
+    unmap(): void;
+    destroy(): void;
+}
+export interface OIDNGPUMappableBuffer extends OIDNGPUBuffer {
+    mapAsync(mode: number, offset?: number, size?: number): Promise<void>;
+}
+export interface OIDNGPUQuerySet {
+    destroy(): void;
+}
+export interface OIDNGPUShaderModule {
+    readonly label: string;
+}
+export interface OIDNGPUBindGroupLayout {
+    readonly label: string;
+}
+export interface OIDNGPUBindGroup {
+    readonly label: string;
+}
+export interface OIDNGPUComputePipeline {
+    getBindGroupLayout(index: number): OIDNGPUBindGroupLayout;
+}
+export interface OIDNGPUCommandBuffer {
+    readonly label: string;
+}
+export interface OIDNGPUBindGroupEntry {
+    binding: number;
+    resource: {
+        buffer: OIDNGPUBuffer;
+        offset?: number;
+        size?: number;
+    };
+}
+export interface OIDNGPUComputePassTimestampWrites {
+    querySet: OIDNGPUQuerySet;
+    beginningOfPassWriteIndex?: number;
+    endOfPassWriteIndex?: number;
+}
+export interface OIDNGPUComputePassEncoder {
+    setPipeline(pipeline: OIDNGPUComputePipeline): void;
+    setBindGroup(index: number, bindGroup: OIDNGPUBindGroup): void;
+    dispatchWorkgroups(x: number, y?: number, z?: number): void;
+    end(): void;
+}
+export interface OIDNGPUCommandEncoder {
+    beginComputePass(descriptor?: {
+        label?: string;
+        timestampWrites?: OIDNGPUComputePassTimestampWrites;
+    }): OIDNGPUComputePassEncoder;
+    copyBufferToBuffer(source: OIDNGPUBuffer, sourceOffset: number, destination: OIDNGPUBuffer, destinationOffset: number, size: number): void;
+    resolveQuerySet(querySet: OIDNGPUQuerySet, firstQuery: number, queryCount: number, destination: OIDNGPUBuffer, destinationOffset: number): void;
+    finish(): OIDNGPUCommandBuffer;
+}
+export interface OIDNGPUQueue {
+    submit(commandBuffers: OIDNGPUCommandBuffer[]): void;
+    writeBuffer(buffer: OIDNGPUBuffer, bufferOffset: number, data: Uint8Array): void;
+}
+export interface OIDNGPUAsyncQueue extends OIDNGPUQueue {
+    onSubmittedWorkDone(): Promise<void>;
+}
+export interface OIDNGPULimits {
+    readonly maxComputeWorkgroupStorageSize: number;
+    readonly maxComputeInvocationsPerWorkgroup: number;
+    readonly maxComputeWorkgroupSizeX: number;
+    readonly maxComputeWorkgroupSizeY: number;
+    readonly maxComputeWorkgroupsPerDimension: number;
+}
+export interface OIDNGPUFeatures {
+    has(name: string): boolean;
+}
+/** Subset of GPUAdapterInfo used to select subgroup-matrix configurations. */
+export interface OIDNGPUAdapterInfo {
+    readonly subgroupMinSize?: number;
+    readonly subgroupMaxSize?: number;
+    readonly subgroupMatrixConfigs?: readonly {
+        componentType: string;
+        resultComponentType: string;
+        M: number;
+        N: number;
+        K: number;
+    }[];
+}
+export interface OIDNGPUDevice {
+    readonly features: OIDNGPUFeatures;
+    readonly limits: OIDNGPULimits;
+    readonly queue: OIDNGPUQueue;
+    createBuffer(descriptor: {
+        label?: string;
+        size: number;
+        usage: number;
+        mappedAtCreation?: boolean;
+    }): OIDNGPUBuffer;
+    createShaderModule(descriptor: {
+        label?: string;
+        code: string;
+    }): OIDNGPUShaderModule;
+    createComputePipeline(descriptor: {
+        label?: string;
+        layout: 'auto';
+        compute: {
+            module: OIDNGPUShaderModule;
+            entryPoint: string;
+        };
+    }): OIDNGPUComputePipeline;
+    createBindGroup(descriptor: {
+        label?: string;
+        layout: OIDNGPUBindGroupLayout;
+        entries: OIDNGPUBindGroupEntry[];
+    }): OIDNGPUBindGroup;
+    createCommandEncoder(descriptor?: {
+        label?: string;
+    }): OIDNGPUCommandEncoder;
+    createQuerySet(descriptor: {
+        label?: string;
+        type: 'timestamp';
+        count: number;
+    }): OIDNGPUQuerySet;
+}
+export interface OIDNGPUAsyncDevice extends OIDNGPUDevice {
+    readonly queue: OIDNGPUAsyncQueue;
+    createComputePipelineAsync(descriptor: {
+        label?: string;
+        layout: 'auto';
+        compute: {
+            module: OIDNGPUShaderModule;
+            entryPoint: string;
+        };
+    }): Promise<OIDNGPUComputePipeline>;
+}