@seanhogg/builderforce-memory-engine 2026.6.18

package/dist/model/mamba_block.js

@@ -0,0 +1,303 @@
+/**
+ * mamba_block.ts – Mamba Mixer Block
+ */
+import { createComputePipeline, createBindGroup, createStorageBuffer, createEmptyStorageBuffer, createUniformBuffer, dispatchKernel, cdiv, } from '../utils/gpu_utils.js';
+import { SELECTIVE_SCAN_FORWARD_WGSL } from '../kernels/selective_scan.js';
+import { gaussianArray } from '../utils/rng.js';
+import { CONV1D_FORWARD_WGSL } from '../kernels/conv1d.js';
+import { LINEAR_FORWARD_WGSL } from '../kernels/linear_projection.js';
+import { ACTIVATIONS_WGSL } from '../kernels/activations.js';
+export class MambaBlock {
+    device;
+    config;
+    dInner;
+    dtRank;
+    wInProj;
+    bInProj;
+    wConv;
+    bConv;
+    wXProj;
+    bXProj;
+    wDtProj;
+    bDtProj;
+    A_log;
+    D_vec;
+    wOutProj;
+    bOutProj;
+    normWeight;
+    gpuWeights;
+    pipelines;
+    _wslaMode = false;
+    constructor(device, config) {
+        this.device = device;
+        this.config = {
+            dState: 16,
+            dConv: 4,
+            expand: 2,
+            biasConv: true,
+            dtRank: Math.ceil(config.dModel / 16),
+            ...config,
+        };
+        const { dModel, expand } = this.config;
+        this.dInner = expand * dModel;
+        this.dtRank = config.dtRank ?? Math.ceil(dModel / 16);
+        // Initialize these before _initWeights so TypeScript is happy
+        this.wInProj = new Float32Array(0);
+        this.bInProj = new Float32Array(0);
+        this.wConv = new Float32Array(0);
+        this.bConv = new Float32Array(0);
+        this.wXProj = new Float32Array(0);
+        this.bXProj = new Float32Array(0);
+        this.wDtProj = new Float32Array(0);
+        this.bDtProj = new Float32Array(0);
+        this.A_log = new Float32Array(0);
+        this.D_vec = new Float32Array(0);
+        this.wOutProj = new Float32Array(0);
+        this.bOutProj = new Float32Array(0);
+        this.normWeight = new Float32Array(0);
+        this.gpuWeights = {};
+        this.pipelines = {};
+        this._initWeights();
+        this._buildPipelines();
+    }
+    _initWeights() {
+        const { dModel, dState, dConv } = this.config;
+        const D = this.dInner;
+        const N = dState;
+        const K = dConv;
+        const R = this.dtRank;
+        const randn = (n, std = 0.02) => gaussianArray(n, std);
+        const zeros = (n) => new Float32Array(n);
+        const ones = (n) => new Float32Array(n).fill(1.0);
+        this.wInProj = randn(2 * D * dModel);
+        this.bInProj = zeros(2 * D);
+        this.wConv = randn(D * K, 0.01);
+        this.bConv = zeros(D);
+        this.wXProj = randn((R + 2 * N) * D, 0.01);
+        this.bXProj = zeros(R + 2 * N);
+        this.wDtProj = randn(D * R, 0.02);
+        this.bDtProj = zeros(D);
+        this.A_log = new Float32Array(D * N);
+        for (let d = 0; d < D; d++) {
+            for (let n = 0; n < N; n++) {
+                this.A_log[d * N + n] = Math.log(n + 1);
+            }
+        }
+        this.D_vec = ones(D);
+        this.wOutProj = randn(dModel * D, 0.02);
+        this.bOutProj = zeros(dModel);
+        this.normWeight = ones(dModel);
+        this._uploadWeightsToGPU();
+    }
+    _uploadWeightsToGPU() {
+        const d = this.device;
+        const mk = (arr, readable = true) => createStorageBuffer(d, arr, readable);
+        this.gpuWeights = {
+            wInProj: mk(this.wInProj),
+            bInProj: mk(this.bInProj),
+            wConv: mk(this.wConv),
+            bConv: mk(this.bConv),
+            wXProj: mk(this.wXProj),
+            bXProj: mk(this.bXProj),
+            wDtProj: mk(this.wDtProj),
+            bDtProj: mk(this.bDtProj),
+            A_log: mk(this.A_log),
+            D_vec: mk(this.D_vec),
+            wOutProj: mk(this.wOutProj),
+            bOutProj: mk(this.bOutProj),
+            normWeight: mk(this.normWeight),
+        };
+    }
+    _buildPipelines() {
+        const d = this.device;
+        this.pipelines = {
+            linear: createComputePipeline(d, LINEAR_FORWARD_WGSL, 'linear_forward'),
+            conv1d: createComputePipeline(d, CONV1D_FORWARD_WGSL, 'conv1d_forward'),
+            silu: createComputePipeline(d, ACTIVATIONS_WGSL, 'silu_forward'),
+            rmsnorm: createComputePipeline(d, ACTIVATIONS_WGSL, 'rmsnorm_forward'),
+            scan_fwd: createComputePipeline(d, SELECTIVE_SCAN_FORWARD_WGSL, 'forward_scan'),
+            scan_reduce: createComputePipeline(d, SELECTIVE_SCAN_FORWARD_WGSL, 'forward_reduce'),
+        };
+    }
+    forward(xBuf, batch, seqLen) {
+        const d = this.device;
+        const { dModel, dState, dConv } = this.config;
+        const D = this.dInner;
+        const N = dState;
+        const B = batch;
+        const L = seqLen;
+        const M = B * L;
+        const R = this.dtRank;
+        const cache = {};
+        const normOut = createEmptyStorageBuffer(d, M * dModel * 4, true);
+        const normInv = createEmptyStorageBuffer(d, M * 4, true);
+        cache.normInv = normInv;
+        cache.normIn = xBuf;
+        {
+            const params = new ArrayBuffer(16);
+            new Uint32Array(params, 0, 2).set([M, dModel]);
+            new Float32Array(params, 8, 1).set([1e-6]);
+            const pBuf = createUniformBuffer(d, params);
+            const bg = createBindGroup(d, this.pipelines['rmsnorm'], [pBuf, xBuf, this.gpuWeights['normWeight'], normOut, normInv]);
+            dispatchKernel(d, this.pipelines['rmsnorm'], bg, [cdiv(M, 64), 1, 1]);
+        }
+        const inProjOut = createEmptyStorageBuffer(d, M * 2 * D * 4, true);
+        cache.normOut = normOut;
+        {
+            const params = new Uint32Array([M, dModel, 2 * D]).buffer;
+            const pBuf = createUniformBuffer(d, params);
+            const bg = createBindGroup(d, this.pipelines['linear'], [pBuf, normOut, this.gpuWeights['wInProj'], this.gpuWeights['bInProj'], inProjOut]);
+            dispatchKernel(d, this.pipelines['linear'], bg, [cdiv(M, 16), cdiv(2 * D, 16), 1]);
+        }
+        const xConvIn = createEmptyStorageBuffer(d, M * D * 4, true);
+        const zBuf = createEmptyStorageBuffer(d, M * D * 4, true);
+        {
+            const enc = d.createCommandEncoder();
+            enc.copyBufferToBuffer(inProjOut, 0, xConvIn, 0, M * D * 4);
+            enc.copyBufferToBuffer(inProjOut, M * D * 4, zBuf, 0, M * D * 4);
+            d.queue.submit([enc.finish()]);
+        }
+        cache.zBuf = zBuf;
+        const convOut = createEmptyStorageBuffer(d, M * D * 4, true);
+        cache.xConvIn = xConvIn;
+        {
+            const params = new Uint32Array([L, D, dConv, B]).buffer;
+            const pBuf = createUniformBuffer(d, params);
+            const bg = createBindGroup(d, this.pipelines['conv1d'], [pBuf, xConvIn, this.gpuWeights['wConv'], this.gpuWeights['bConv'], convOut]);
+            dispatchKernel(d, this.pipelines['conv1d'], bg, [cdiv(L, 16), cdiv(D, 16), B]);
+        }
+        const siluOut = createEmptyStorageBuffer(d, M * D * 4, true);
+        cache.convOut = convOut;
+        {
+            const params = new Uint32Array([M * D]).buffer;
+            const pBuf = createUniformBuffer(d, params);
+            const bg = createBindGroup(d, this.pipelines['silu'], [pBuf, convOut, siluOut]);
+            dispatchKernel(d, this.pipelines['silu'], bg, [cdiv(M * D, 256), 1, 1]);
+        }
+        const xProjOut = createEmptyStorageBuffer(d, M * (R + 2 * N) * 4, true);
+        {
+            const params = new Uint32Array([M, D, R + 2 * N]).buffer;
+            const pBuf = createUniformBuffer(d, params);
+            const bg = createBindGroup(d, this.pipelines['linear'], [pBuf, siluOut, this.gpuWeights['wXProj'], this.gpuWeights['bXProj'], xProjOut]);
+            dispatchKernel(d, this.pipelines['linear'], bg, [cdiv(M, 16), cdiv(R + 2 * N, 16), 1]);
+        }
+        const dtRaw = createEmptyStorageBuffer(d, M * R * 4, true);
+        const B_raw = createEmptyStorageBuffer(d, B * L * N * 4, true);
+        const C_raw = createEmptyStorageBuffer(d, B * L * N * 4, true);
+        {
+            const enc = d.createCommandEncoder();
+            enc.copyBufferToBuffer(xProjOut, 0, dtRaw, 0, M * R * 4);
+            enc.copyBufferToBuffer(xProjOut, M * R * 4, B_raw, 0, B * L * N * 4);
+            enc.copyBufferToBuffer(xProjOut, M * (R + N) * 4, C_raw, 0, B * L * N * 4);
+            d.queue.submit([enc.finish()]);
+        }
+        const deltaFull = createEmptyStorageBuffer(d, M * D * 4, true);
+        {
+            const params = new Uint32Array([M, R, D]).buffer;
+            const pBuf = createUniformBuffer(d, params);
+            const bg = createBindGroup(d, this.pipelines['linear'], [pBuf, dtRaw, this.gpuWeights['wDtProj'], this.gpuWeights['bDtProj'], deltaFull]);
+            dispatchKernel(d, this.pipelines['linear'], bg, [cdiv(M, 16), cdiv(D, 16), 1]);
+        }
+        const scanY = createEmptyStorageBuffer(d, B * L * D * 4, true);
+        const hCache = createEmptyStorageBuffer(d, 2 * B * L * D * N * 4, true);
+        cache.siluOut = siluOut;
+        cache.deltaFull = deltaFull;
+        cache.B_raw = B_raw;
+        cache.C_raw = C_raw;
+        cache.hCache = hCache;
+        {
+            const params = new Uint32Array([L, N, D, B]).buffer;
+            const pBuf = createUniformBuffer(d, params);
+            const bg = createBindGroup(d, this.pipelines['scan_fwd'], [pBuf, siluOut, deltaFull, this.gpuWeights['A_log'], B_raw, C_raw,
+                this.gpuWeights['D_vec'], scanY, hCache]);
+            dispatchKernel(d, this.pipelines['scan_fwd'], bg, [cdiv(D, 8), cdiv(N, 8), B]);
+            const bg2 = createBindGroup(d, this.pipelines['scan_reduce'], [pBuf, siluOut, deltaFull, this.gpuWeights['A_log'], B_raw, C_raw,
+                this.gpuWeights['D_vec'], scanY, hCache]);
+            dispatchKernel(d, this.pipelines['scan_reduce'], bg2, [cdiv(L, 64), D, B]);
+        }
+        const siluZ = createEmptyStorageBuffer(d, M * D * 4, true);
+        const gatedOut = createEmptyStorageBuffer(d, M * D * 4, true);
+        {
+            const params = new Uint32Array([M * D]).buffer;
+            const pBuf = createUniformBuffer(d, params);
+            const bg = createBindGroup(d, this.pipelines['silu'], [pBuf, zBuf, siluZ]);
+            dispatchKernel(d, this.pipelines['silu'], bg, [cdiv(M * D, 256), 1, 1]);
+            const mulShader = /* wgsl */ `
+                @group(0) @binding(0) var<storage, read>       a : array<f32>;
+                @group(0) @binding(1) var<storage, read>       b : array<f32>;
+                @group(0) @binding(2) var<storage, read_write> c : array<f32>;
+                @group(0) @binding(3) var<uniform>             n : u32;
+                @compute @workgroup_size(256)
+                fn main(@builtin(global_invocation_id) gid: vec3<u32>) {
+                    let i = gid.x;
+                    if (i < n) { c[i] = a[i] * b[i]; }
+                }
+            `;
+            const mulPipeline = createComputePipeline(d, mulShader, 'main');
+            const nBuf = createUniformBuffer(d, new Uint32Array([M * D]).buffer);
+            const bgMul = createBindGroup(d, mulPipeline, [scanY, siluZ, gatedOut, nBuf]);
+            dispatchKernel(d, mulPipeline, bgMul, [cdiv(M * D, 256), 1, 1]);
+        }
+        const outProjOut = createEmptyStorageBuffer(d, M * dModel * 4, true);
+        {
+            const params = new Uint32Array([M, D, dModel]).buffer;
+            const pBuf = createUniformBuffer(d, params);
+            const bg = createBindGroup(d, this.pipelines['linear'], [pBuf, gatedOut, this.gpuWeights['wOutProj'], this.gpuWeights['bOutProj'], outProjOut]);
+            dispatchKernel(d, this.pipelines['linear'], bg, [cdiv(M, 16), cdiv(dModel, 16), 1]);
+        }
+        const output = createEmptyStorageBuffer(d, M * dModel * 4, true);
+        {
+            const addShader = /* wgsl */ `
+                @group(0) @binding(0) var<storage, read>       a : array<f32>;
+                @group(0) @binding(1) var<storage, read>       b : array<f32>;
+                @group(0) @binding(2) var<storage, read_write> c : array<f32>;
+                @group(0) @binding(3) var<uniform>             n : u32;
+                @compute @workgroup_size(256)
+                fn main(@builtin(global_invocation_id) gid: vec3<u32>) {
+                    let i = gid.x;
+                    if (i < n) { c[i] = a[i] + b[i]; }
+                }
+            `;
+            const addPipeline = createComputePipeline(d, addShader, 'main');
+            const nBuf = createUniformBuffer(d, new Uint32Array([M * dModel]).buffer);
+            const bgAdd = createBindGroup(d, addPipeline, [outProjOut, xBuf, output, nBuf]);
+            dispatchKernel(d, addPipeline, bgAdd, [cdiv(M * dModel, 256), 1, 1]);
+        }
+        return { output, cache };
+    }
+    parameters() {
+        const { dModel, dState, dConv } = this.config;
+        const D = this.dInner;
+        const N = dState;
+        const K = dConv;
+        const R = this.dtRank;
+        return [
+            { buf: this.gpuWeights['wInProj'], numel: 2 * D * dModel, name: 'wInProj' },
+            { buf: this.gpuWeights['bInProj'], numel: 2 * D, name: 'bInProj' },
+            { buf: this.gpuWeights['wConv'], numel: D * K, name: 'wConv' },
+            { buf: this.gpuWeights['bConv'], numel: D, name: 'bConv' },
+            { buf: this.gpuWeights['wXProj'], numel: (R + 2 * N) * D, name: 'wXProj' },
+            { buf: this.gpuWeights['bXProj'], numel: R + 2 * N, name: 'bXProj' },
+            { buf: this.gpuWeights['wDtProj'], numel: D * R, name: 'wDtProj' },
+            { buf: this.gpuWeights['bDtProj'], numel: D, name: 'bDtProj' },
+            { buf: this.gpuWeights['A_log'], numel: D * N, name: 'A_log' },
+            { buf: this.gpuWeights['D_vec'], numel: D, name: 'D_vec' },
+            { buf: this.gpuWeights['wOutProj'], numel: dModel * D, name: 'wOutProj' },
+            { buf: this.gpuWeights['bOutProj'], numel: dModel, name: 'bOutProj' },
+            { buf: this.gpuWeights['normWeight'], numel: dModel, name: 'normWeight' },
+        ];
+    }
+    setWSLAMode(enabled) {
+        this._wslaMode = enabled;
+    }
+    getTrainableParams() {
+        if (this._wslaMode) {
+            return [
+                { buf: this.gpuWeights['wXProj'], numel: this.wXProj.length, name: 'wXProj' },
+                { buf: this.gpuWeights['bXProj'], numel: this.bXProj.length, name: 'bXProj' },
+            ];
+        }
+        return this.parameters();
+    }
+}
+//# sourceMappingURL=mamba_block.js.map

package/dist/model/mamba_block.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mamba_block.js","sourceRoot":"","sources":["../../src/model/mamba_block.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EACH,qBAAqB,EACrB,eAAe,EACf,mBAAmB,EACnB,wBAAwB,EACxB,mBAAmB,EACnB,cAAc,EACd,IAAI,GACP,MAAM,uBAAuB,CAAC;AAE/B,OAAO,EAAE,2BAA2B,EAAE,MAAO,8BAA8B,CAAC;AAC5E,OAAO,EAAE,aAAa,EAAE,MAAqB,iBAAiB,CAAC;AAC/D,OAAO,EAAE,mBAAmB,EAAE,MAAe,sBAAsB,CAAC;AACpE,OAAO,EAAE,mBAAmB,EAAE,MAAe,iCAAiC,CAAC;AAC/E,OAAO,EAAE,gBAAgB,EAAE,MAAkB,2BAA2B,CAAC;AAoCzE,MAAM,OAAO,UAAU;IACnB,MAAM,CAAY;IAClB,MAAM,CAA6B;IACnC,MAAM,CAAS;IACf,MAAM,CAAS;IACf,OAAO,CAAe;IACtB,OAAO,CAAe;IACtB,KAAK,CAAe;IACpB,KAAK,CAAe;IACpB,MAAM,CAAe;IACrB,MAAM,CAAe;IACrB,OAAO,CAAe;IACtB,OAAO,CAAe;IACtB,KAAK,CAAe;IACpB,KAAK,CAAe;IACpB,QAAQ,CAAe;IACvB,QAAQ,CAAe;IACvB,UAAU,CAAe;IACzB,UAAU,CAA4B;IACtC,SAAS,CAAqC;IACtC,SAAS,GAAG,KAAK,CAAC;IAE1B,YAAY,MAAiB,EAAE,MAAwB;QACnD,IAAI,CAAC,MAAM,GAAI,MAAM,CAAC;QACtB,IAAI,CAAC,MAAM,GAAI;YACX,MAAM,EAAI,EAAE;YACZ,KAAK,EAAK,CAAC;YACX,MAAM,EAAI,CAAC;YACX,QAAQ,EAAE,IAAI;YACd,MAAM,EAAI,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,GAAG,EAAE,CAAC;YACvC,GAAG,MAAM;SACkB,CAAC;QAEhC,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,IAAI,CAAC,MAAM,CAAC;QACvC,IAAI,CAAC,MAAM,GAAI,MAAM,GAAG,MAAM,CAAC;QAC/B,IAAI,CAAC,MAAM,GAAI,MAAM,CAAC,MAAM,IAAI,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,EAAE,CAAC,CAAC;QAEvD,8DAA8D;QAC9D,IAAI,CAAC,OAAO,GAAG,IAAI,YAAY,CAAC,CAAC,CAAC,CAAC;QACnC,IAAI,CAAC,OAAO,GAAG,IAAI,YAAY,CAAC,CAAC,CAAC,CAAC;QACnC,IAAI,CAAC,KAAK,GAAG,IAAI,YAAY,CAAC,CAAC,CAAC,CAAC;QACjC,IAAI,CAAC,KAAK,GAAG,IAAI,YAAY,CAAC,CAAC,CAAC,CAAC;QACjC,IAAI,CAAC,MAAM,GAAG,IAAI,YAAY,CAAC,CAAC,CAAC,CAAC;QAClC,IAAI,CAAC,MAAM,GAAG,IAAI,YAAY,CAAC,CAAC,CAAC,CAAC;QAClC,IAAI,CAAC,OAAO,GAAG,IAAI,YAAY,CAAC,CAAC,CAAC,CAAC;QACnC,IAAI,CAAC,OAAO,GAAG,IAAI,YAAY,CAAC,CAAC,CAAC,CAAC;QACnC,IAAI,CAAC,KAAK,GAAG,IAAI,YAAY,CAAC,CAAC,CAAC,CAAC;QACjC,IAAI,CAAC,KAAK,GAAG,IAAI,YAAY,CAAC,CAAC,CAAC,CAAC;QACjC,IAAI,CAAC,QAAQ,GAAG,IAAI,YAAY,CAAC,CAAC,CAAC,CAAC;QACpC,IAAI,CAAC,QAAQ,GAAG,IAAI,YAAY,CAAC,CAAC,CAAC,CAAC;QACpC,IAAI,CAAC,UAAU,GAAG,IAAI,YAAY,CAAC,CAAC,CAAC,CAAC;QACtC,IAAI,CAAC,UAAU,GAAG,EAAE,CAAC;QACrB,IAAI,CAAC,SAAS,GAAG,EAAE,CAAC;QAEpB,IAAI,CAAC,YAAY,EAAE,CAAC;QACpB,IAAI,CAAC,eAAe,EAAE,CAAC;IAC3B,CAAC;IAEO,YAAY;QAChB,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,IAAI,CAAC,MAAM,CAAC;QAC9C,MAAM,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC;QACtB,MAAM,CAAC,GAAG,MAAM,CAAC;QACjB,MAAM,CAAC,GAAG,KAAK,CAAC;QAChB,MAAM,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC;QAEtB,MAAM,KAAK,GAAG,CAAC,CAAS,EAAE,GAAG,GAAG,IAAI,EAAgB,EAAE,CAAC,aAAa,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;QAE7E,MAAM,KAAK,GAAI,CAAC,CAAS,EAAmB,EAAE,CAAC,IAAI,YAAY,CAAC,CAAC,CAAC,CAAC;QACnE,MAAM,IAAI,GAAK,CAAC,CAAS,EAAmB,EAAE,CAAC,IAAI,YAAY,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAE7E,IAAI,CAAC,OAAO,GAAI,KAAK,CAAC,CAAC,GAAG,CAAC,GAAG,MAAM,CAAC,CAAC;QACtC,IAAI,CAAC,OAAO,GAAI,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;QAC7B,IAAI,CAAC,KAAK,GAAM,KAAK,CAAC,CAAC,GAAG,CAAC,EAAE,IAAI,CAAC,CAAC;QACnC,IAAI,CAAC,KAAK,GAAM,KAAK,CAAC,CAAC,CAAC,CAAC;QACzB,IAAI,CAAC,MAAM,GAAK,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,EAAE,IAAI,CAAC,CAAC;QAC7C,IAAI,CAAC,MAAM,GAAK,KAAK,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC;QACjC,IAAI,CAAC,OAAO,GAAI,KAAK,CAAC,CAAC,GAAG,CAAC,EAAE,IAAI,CAAC,CAAC;QACnC,IAAI,CAAC,OAAO,GAAI,KAAK,CAAC,CAAC,CAAC,CAAC;QAEzB,IAAI,CAAC,KAAK,GAAM,IAAI,YAAY,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;QACxC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;YACzB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;gBACzB,IAAI,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;YAC5C,CAAC;QACL,CAAC;QAED,IAAI,CAAC,KAAK,GAAM,IAAI,CAAC,CAAC,CAAC,CAAC;QACxB,IAAI,CAAC,QAAQ,GAAG,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,IAAI,CAAC,CAAC;QACxC,IAAI,CAAC,QAAQ,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC;QAC9B,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC;QAE/B,IAAI,CAAC,mBAAmB,EAAE,CAAC;IAC/B,CAAC;IAEO,mBAAmB;QACvB,MAAM,CAAC,GAAI,IAAI,CAAC,MAAM,CAAC;QACvB,MAAM,EAAE,GAAG,CAAC,GAAiB,EAAE,QAAQ,GAAG,IAAI,EAAa,EAAE,CAAC,mBAAmB,CAAC,CAAC,EAAE,GAAG,EAAE,QAAQ,CAAC,CAAC;QAEpG,IAAI,CAAC,UAAU,GAAG;YACd,OAAO,EAAI,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC;YAC3B,OAAO,EAAI,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC;YAC3B,KAAK,EAAM,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC;YACzB,KAAK,EAAM,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC;YACzB,MAAM,EAAK,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC;YAC1B,MAAM,EAAK,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC;YAC1B,OAAO,EAAI,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC;YAC3B,OAAO,EAAI,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC;YAC3B,KAAK,EAAM,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC;YACzB,KAAK,EAAM,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC;YACzB,QAAQ,EAAG,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC;YAC5B,QAAQ,EAAG,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC;YAC5B,UAAU,EAAE,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC;SAClC,CAAC;IACN,CAAC;IAEO,eAAe;QACnB,MAAM,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC;QAEtB,IAAI,CAAC,SAAS,GAAG;YACb,MAAM,EAAM,qBAAqB,CAAC,CAAC,EAAE,mBAAmB,EAAY,gBAAgB,CAAC;YACrF,MAAM,EAAM,qBAAqB,CAAC,CAAC,EAAE,mBAAmB,EAAY,gBAAgB,CAAC;YACrF,IAAI,EAAQ,qBAAqB,CAAC,CAAC,EAAE,gBAAgB,EAAe,cAAc,CAAC;YACnF,OAAO,EAAK,qBAAqB,CAAC,CAAC,EAAE,gBAAgB,EAAe,iBAAiB,CAAC;YACtF,QAAQ,EAAI,qBAAqB,CAAC,CAAC,EAAE,2BAA2B,EAAI,cAAc,CAAC;YACnF,WAAW,EAAE,qBAAqB,CAAC,CAAC,EAAE,2BAA2B,EAAG,gBAAgB,CAAC;SACxF,CAAC;IACN,CAAC;IAED,OAAO,CAAC,IAAe,EAAE,KAAa,EAAE,MAAc;QAClD,MAAM,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC;QACtB,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,IAAI,CAAC,MAAM,CAAC;QAC9C,MAAM,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC;QACtB,MAAM,CAAC,GAAG,MAAM,CAAC;QACjB,MAAM,CAAC,GAAG,KAAK,CAAC;QAChB,MAAM,CAAC,GAAG,MAAM,CAAC;QACjB,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QAChB,MAAM,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC;QAEtB,MAAM,KAAK,GAAG,EAAgB,CAAC;QAE/B,MAAM,OAAO,GAAI,wBAAwB,CAAC,CAAC,EAAE,CAAC,GAAG,MAAM,GAAG,CAAC,EAAE,IAAI,CAAC,CAAC;QACnE,MAAM,OAAO,GAAI,wBAAwB,CAAC,CAAC,EAAE,CAAC,GAAG,CAAC,EAAW,IAAI,CAAC,CAAC;QACnE,KAAK,CAAC,OAAO,GAAI,OAAO,CAAC;QACzB,KAAK,CAAC,MAAM,GAAK,IAAI,CAAC;QAEtB,CAAC;YACG,MAAM,MAAM,GAAG,IAAI,WAAW,CAAC,EAAE,CAAC,CAAC;YACnC,IAAI,WAAW,CAAC,MAAM,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,CAAC;YAC/C,IAAI,YAAY,CAAC,MAAM,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;YAC3C,MAAM,IAAI,GAAG,mBAAmB,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC;YAE5C,MAAM,EAAE,GAAG,eAAe,CAAC,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,SAAS,CAAE,EACpD,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,UAAU,CAAC,YAAY,CAAE,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;YACpE,cAAc,CAAC,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,SAAS,CAAE,EAAE,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QAC3E,CAAC;QAED,MAAM,SAAS,GAAG,wBAAwB,CAAC,CAAC,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,IAAI,CAAC,CAAC;QACnE,KAAK,CAAC,OAAO,GAAK,OAAO,CAAC;QAC1B,CAAC;YACG,MAAM,MAAM,GAAG,IAAI,WAAW,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC;YAC1D,MAAM,IAAI,GAAK,mBAAmB,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC;YAC9C,MAAM,EAAE,GAAG,eAAe,CAAC,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAE,EACnD,CAAC,IAAI,EAAE,OAAO,EAAE,IAAI,CAAC,UAAU,CAAC,SAAS,CAAE,EAAE,IAAI,CAAC,UAAU,CAAC,SAAS,CAAE,EAAE,SAAS,CAAC,CAAC,CAAC;YAC1F,cAAc,CAAC,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAE,EAAE,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,IAAI,CAAC,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QACxF,CAAC;QAED,MAAM,OAAO,GAAI,wBAAwB,CAAC,CAAC,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,IAAI,CAAC,CAAC;QAC9D,MAAM,IAAI,GAAO,wBAAwB,CAAC,CAAC,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,IAAI,CAAC,CAAC;QAC9D,CAAC;YACG,MAAM,GAAG,GAAG,CAAC,CAAC,oBAAoB,EAAE,CAAC;YACrC,GAAG,CAAC,kBAAkB,CAAC,SAAS,EAAE,CAAC,EAAY,OAAO,EAAE,CAAC,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC;YACtE,GAAG,CAAC,kBAAkB,CAAC,SAAS,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC,EAAI,IAAI,EAAK,CAAC,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC;YACtE,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QACnC,CAAC;QACD,KAAK,CAAC,IAAI,GAAG,IAAI,CAAC;QAElB,MAAM,OAAO,GAAG,wBAAwB,CAAC,CAAC,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,IAAI,CAAC,CAAC;QAC7D,KAAK,CAAC,OAAO,GAAG,OAAO,CAAC;QACxB,CAAC;YACG,MAAM,MAAM,GAAG,IAAI,WAAW,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC;YACxD,MAAM,IAAI,GAAK,mBAAmB,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC;YAC9C,MAAM,EAAE,GAAG,eAAe,CAAC,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAE,EACnD,CAAC,IAAI,EAAE,OAAO,EAAE,IAAI,CAAC,UAAU,CAAC,OAAO,CAAE,EAAE,IAAI,CAAC,UAAU,CAAC,OAAO,CAAE,EAAE,OAAO,CAAC,CAAC,CAAC;YACpF,cAAc,CAAC,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAE,EAAE,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,IAAI,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QACpF,CAAC;QAED,MAAM,OAAO,GAAG,wBAAwB,CAAC,CAAC,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,IAAI,CAAC,CAAC;QAC7D,KAAK,CAAC,OAAO,GAAG,OAAO,CAAC;QACxB,CAAC;YACG,MAAM,MAAM,GAAG,IAAI,WAAW,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC;YAC/C,MAAM,IAAI,GAAK,mBAAmB,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC;YAC9C,MAAM,EAAE,GAAG,eAAe,CAAC,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAE,EACjD,CAAC,IAAI,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;YAC9B,cAAc,CAAC,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAE,EAAE,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,EAAE,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QAC7E,CAAC;QAED,MAAM,QAAQ,GAAG,wBAAwB,CAAC,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,EAAE,IAAI,CAAC,CAAC;QACxE,CAAC;YACG,MAAM,MAAM,GAAG,IAAI,WAAW,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC;YACzD,MAAM,IAAI,GAAK,mBAAmB,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC;YAC9C,MAAM,EAAE,GAAG,eAAe,CAAC,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAE,EACnD,CAAC,IAAI,EAAE,OAAO,EAAE,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAE,EAAE,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAE,EAAE,QAAQ,CAAC,CAAC,CAAC;YACvF,cAAc,CAAC,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAE,EAAE,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,IAAI,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QAC5F,CAAC;QAED,MAAM,KAAK,GAAG,wBAAwB,CAAC,CAAC,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC,EAAM,IAAI,CAAC,CAAC;QAC/D,MAAM,KAAK,GAAG,wBAAwB,CAAC,CAAC,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,IAAI,CAAC,CAAC;QAC/D,MAAM,KAAK,GAAG,wBAAwB,CAAC,CAAC,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,IAAI,CAAC,CAAC;QAC/D,CAAC;YACG,MAAM,GAAG,GAAG,CAAC,CAAC,oBAAoB,EAAE,CAAC;YACrC,GAAG,CAAC,kBAAkB,CAAC,QAAQ,EAAE,CAAC,EAAmB,KAAK,EAAE,CAAC,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC;YAC1E,GAAG,CAAC,kBAAkB,CAAC,QAAQ,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC,EAAW,KAAK,EAAE,CAAC,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC;YAC9E,GAAG,CAAC,kBAAkB,CAAC,QAAQ,EAAE,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,EAAK,KAAK,EAAE,CAAC,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC;YAC9E,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QACnC,CAAC;QAED,MAAM,SAAS,GAAG,wBAAwB,CAAC,CAAC,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,IAAI,CAAC,CAAC;QAC/D,CAAC;YACG,MAAM,MAAM,GAAG,IAAI,WAAW,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC;YACjD,MAAM,IAAI,GAAK,mBAAmB,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC;YAC9C,MAAM,EAAE,GAAG,eAAe,CAAC,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAE,EACnD,CAAC,IAAI,EAAE,KAAK,EAAE,IAAI,CAAC,UAAU,CAAC,SAAS,CAAE,EAAE,IAAI,CAAC,UAAU,CAAC,SAAS,CAAE,EAAE,SAAS,CAAC,CAAC,CAAC;YACxF,cAAc,CAAC,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAE,EAAE,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,IAAI,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QACpF,CAAC;QAED,MAAM,KAAK,GAAQ,wBAAwB,CAAC,CAAC,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,EAAU,IAAI,CAAC,CAAC;QAC5E,MAAM,MAAM,GAAO,wBAAwB,CAAC,CAAC,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,IAAI,CAAC,CAAC;QAC5E,KAAK,CAAC,OAAO,GAAM,OAAO,CAAC;QAC3B,KAAK,CAAC,SAAS,GAAI,SAAS,CAAC;QAC7B,KAAK,CAAC,KAAK,GAAQ,KAAK,CAAC;QACzB,KAAK,CAAC,KAAK,GAAQ,KAAK,CAAC;QACzB,KAAK,CAAC,MAAM,GAAO,MAAM,CAAC;QAE1B,CAAC;YACG,MAAM,MAAM,GAAG,IAAI,WAAW,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC;YACpD,MAAM,IAAI,GAAK,mBAAmB,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC;YAE9C,MAAM,EAAE,GAAG,eAAe,CAAC,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,UAAU,CAAE,EACrD,CAAC,IAAI,EAAE,OAAO,EAAE,SAAS,EAAE,IAAI,CAAC,UAAU,CAAC,OAAO,CAAE,EAAE,KAAK,EAAE,KAAK;gBACjE,IAAI,CAAC,UAAU,CAAC,OAAO,CAAE,EAAE,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC;YAChD,cAAc,CAAC,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,UAAU,CAAE,EAAE,EAAE,EAC7C,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;YAEjC,MAAM,GAAG,GAAG,eAAe,CAAC,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,aAAa,CAAE,EACzD,CAAC,IAAI,EAAE,OAAO,EAAE,SAAS,EAAE,IAAI,CAAC,UAAU,CAAC,OAAO,CAAE,EAAE,KAAK,EAAE,KAAK;gBACjE,IAAI,CAAC,UAAU,CAAC,OAAO,CAAE,EAAE,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC;YAChD,cAAc,CAAC,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,aAAa,CAAE,EAAE,GAAG,EACjD,CAAC,IAAI,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QAC7B,CAAC;QAED,MAAM,KAAK,GAAK,wBAAwB,CAAC,CAAC,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,IAAI,CAAC,CAAC;QAC7D,MAAM,QAAQ,GAAG,wBAAwB,CAAC,CAAC,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,IAAI,CAAC,CAAC;QAC9D,CAAC;YACG,MAAM,MAAM,GAAG,IAAI,WAAW,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC;YAC/C,MAAM,IAAI,GAAK,mBAAmB,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC;YAC9C,MAAM,EAAE,GAAG,eAAe,CAAC,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAE,EACjD,CAAC,IAAI,EAAE,IAAI,EAAE,KAAK,CAAC,CAAC,CAAC;YACzB,cAAc,CAAC,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAE,EAAE,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,EAAE,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;YAEzE,MAAM,SAAS,GAAG,UAAU,CAAA;;;;;;;;;;aAU3B,CAAC;YACF,MAAM,WAAW,GAAG,qBAAqB,CAAC,CAAC,EAAE,SAAS,EAAE,MAAM,CAAC,CAAC;YAChE,MAAM,IAAI,GAAG,mBAAmB,CAAC,CAAC,EAAE,IAAI,WAAW,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC;YACrE,MAAM,KAAK,GAAG,eAAe,CAAC,CAAC,EAAE,WAAW,EACxC,CAAC,KAAK,EAAE,KAAK,EAAE,QAAQ,EAAE,IAAI,CAAC,CAAC,CAAC;YACpC,cAAc,CAAC,CAAC,EAAE,WAAW,EAAE,KAAK,EAAE,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,EAAE,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QACpE,CAAC;QAED,MAAM,UAAU,GAAG,wBAAwB,CAAC,CAAC,EAAE,CAAC,GAAG,MAAM,GAAG,CAAC,EAAE,IAAI,CAAC,CAAC;QACrE,CAAC;YACG,MAAM,MAAM,GAAG,IAAI,WAAW,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC;YACtD,MAAM,IAAI,GAAK,mBAAmB,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC;YAC9C,MAAM,EAAE,GAAG,eAAe,CAAC,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAE,EACnD,CAAC,IAAI,EAAE,QAAQ,EAAE,IAAI,CAAC,UAAU,CAAC,UAAU,CAAE,EAAE,IAAI,CAAC,UAAU,CAAC,UAAU,CAAE,EAAE,UAAU,CAAC,CAAC,CAAC;YAC9F,cAAc,CAAC,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAE,EAAE,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,IAAI,CAAC,MAAM,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QACzF,CAAC;QAED,MAAM,MAAM,GAAG,wBAAwB,CAAC,CAAC,EAAE,CAAC,GAAG,MAAM,GAAG,CAAC,EAAE,IAAI,CAAC,CAAC;QACjE,CAAC;YACG,MAAM,SAAS,GAAG,UAAU,CAAA;;;;;;;;;;aAU3B,CAAC;YACF,MAAM,WAAW,GAAG,qBAAqB,CAAC,CAAC,EAAE,SAAS,EAAE,MAAM,CAAC,CAAC;YAChE,MAAM,IAAI,GAAG,mBAAmB,CAAC,CAAC,EAAE,IAAI,WAAW,CAAC,CAAC,CAAC,GAAG,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC;YAC1E,MAAM,KAAK,GAAG,eAAe,CAAC,CAAC,EAAE,WAAW,EACxC,CAAC,UAAU,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,CAAC,CAAC,CAAC;YACtC,cAAc,CAAC,CAAC,EAAE,WAAW,EAAE,KAAK,EAAE,CAAC,IAAI,CAAC,CAAC,GAAG,MAAM,EAAE,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QACzE,CAAC;QAED,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC;IAC7B,CAAC;IAED,UAAU;QACN,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,IAAI,CAAC,MAAM,CAAC;QAC9C,MAAM,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC;QACtB,MAAM,CAAC,GAAG,MAAM,CAAC;QACjB,MAAM,CAAC,GAAG,KAAK,CAAC;QAChB,MAAM,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC;QAEtB,OAAO;YACH,EAAE,GAAG,EAAE,IAAI,CAAC,UAAU,CAAC,SAAS,CAAE,EAAI,KAAK,EAAE,CAAC,GAAG,CAAC,GAAG,MAAM,EAAE,IAAI,EAAE,SAAS,EAAI;YAChF,EAAE,GAAG,EAAE,IAAI,CAAC,UAAU,CAAC,SAAS,CAAE,EAAI,KAAK,EAAE,CAAC,GAAG,CAAC,EAAW,IAAI,EAAE,SAAS,EAAI;YAChF,EAAE,GAAG,EAAE,IAAI,CAAC,UAAU,CAAC,OAAO,CAAE,EAAM,KAAK,EAAE,CAAC,GAAG,CAAC,EAAY,IAAI,EAAE,OAAO,EAAM;YACjF,EAAE,GAAG,EAAE,IAAI,CAAC,UAAU,CAAC,OAAO,CAAE,EAAM,KAAK,EAAE,CAAC,EAAgB,IAAI,EAAE,OAAO,EAAM;YACjF,EAAE,GAAG,EAAE,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAE,EAAK,KAAK,EAAE,CAAC,CAAC,GAAG,CAAC,GAAC,CAAC,CAAC,GAAG,CAAC,EAAI,IAAI,EAAE,QAAQ,EAAI;YAChF,EAAE,GAAG,EAAE,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAE,EAAK,KAAK,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC,EAAQ,IAAI,EAAE,QAAQ,EAAI;YAChF,EAAE,GAAG,EAAE,IAAI,CAAC,UAAU,CAAC,SAAS,CAAE,EAAI,KAAK,EAAE,CAAC,GAAG,CAAC,EAAY,IAAI,EAAE,SAAS,EAAG;YAChF,EAAE,GAAG,EAAE,IAAI,CAAC,UAAU,CAAC,SAAS,CAAE,EAAI,KAAK,EAAE,CAAC,EAAgB,IAAI,EAAE,SAAS,EAAG;YAChF,EAAE,GAAG,EAAE,IAAI,CAAC,UAAU,CAAC,OAAO,CAAE,EAAM,KAAK,EAAE,CAAC,GAAG,CAAC,EAAY,IAAI,EAAE,OAAO,EAAK;YAChF,EAAE,GAAG,EAAE,IAAI,CAAC,UAAU,CAAC,OAAO,CAAE,EAAM,KAAK,EAAE,CAAC,EAAgB,IAAI,EAAE,OAAO,EAAK;YAChF,EAAE,GAAG,EAAE,IAAI,CAAC,UAAU,CAAC,UAAU,CAAE,EAAG,KAAK,EAAE,MAAM,GAAG,CAAC,EAAO,IAAI,EAAE,UAAU,EAAE;YAChF,EAAE,GAAG,EAAE,IAAI,CAAC,UAAU,CAAC,UAAU,CAAE,EAAG,KAAK,EAAE,MAAM,EAAW,IAAI,EAAE,UAAU,EAAE;YAChF,EAAE,GAAG,EAAE,IAAI,CAAC,UAAU,CAAC,YAAY,CAAE,EAAE,KAAK,EAAE,MAAM,EAAW,IAAI,EAAE,YAAY,EAAC;SACrF,CAAC;IACN,CAAC;IAED,WAAW,CAAC,OAAgB;QACxB,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC;IAC7B,CAAC;IAED,kBAAkB;QACd,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACjB,OAAO;gBACH,EAAE,GAAG,EAAE,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAE,EAAE,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,IAAI,EAAE,QAAQ,EAAE;gBAC9E,EAAE,GAAG,EAAE,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAE,EAAE,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,IAAI,EAAE,QAAQ,EAAE;aACjF,CAAC;QACN,CAAC;QACD,OAAO,IAAI,CAAC,UAAU,EAAE,CAAC;IAC7B,CAAC;CACJ"}

package/dist/model/mamba_model.d.ts

@@ -0,0 +1,140 @@
+/**
+ * mamba_model.ts – HybridMambaModel: Mamba-1/2/3 and Attention layer scheduling.
+ *
+ * Replaces the fixed MambaBlock[] array with a SequenceLayer[] built from a
+ * per-layer type schedule.  MambaModel is kept as a backward-compatible alias
+ * (all-mamba1 schedule).
+ *
+ * MBJS binary format:
+ *   Version 1 (legacy): [magic][v=1][nParams][numel[]][ f32 data ]
+ *   Version 2 (new):    [magic][v=2][nLayers][layerType[]][padding][nParams][numel[]][ f32 data ]
+ *     layerType: 0=mamba1, 1=mamba2, 2=mamba3, 3=attention
+ */
+import type { SequenceLayer, LayerParam, LayerType } from './sequence_layer.js';
+import type { Mamba1BlockConfig } from './mamba1_block.js';
+import type { Mamba2BlockConfig } from './mamba2_block.js';
+import type { Mamba3BlockConfig } from './mamba3_block.js';
+import type { AttentionBlockConfig } from './attention_block.js';
+export interface LayerSpec {
+    type: LayerType;
+    config?: Partial<Mamba1BlockConfig | Mamba2BlockConfig | Mamba3BlockConfig | AttentionBlockConfig>;
+}
+export interface HybridMambaModelConfig {
+    vocabSize: number;
+    dModel: number;
+    numLayers: number;
+    /**
+     * Per-layer type schedule.  Length must equal numLayers.
+     * Defaults to all 'mamba1' (backward-compatible).
+     */
+    layers?: LayerSpec[];
+    defaultMamba1?: Partial<Mamba1BlockConfig>;
+    defaultMamba2?: Partial<Mamba2BlockConfig>;
+    defaultMamba3?: Partial<Mamba3BlockConfig>;
+    defaultAttention?: Partial<AttentionBlockConfig>;
+    dState?: number;
+    dConv?: number;
+    expand?: number;
+    nHeads?: number;
+    nGroups?: number;
+    chunkLen?: number;
+    mimoGroup?: number;
+    eosId?: number;
+    /**
+     * Optional deterministic seed for weight initialisation. When set, the
+     * embedding table and all block weights are initialised reproducibly — the
+     * same seed yields byte-identical initial weights on any machine. When
+     * omitted, weights use `Math.random` (non-reproducible) as before.
+     */
+    seed?: number;
+}
+/** Legacy Mamba-1-only config (fully backward-compatible). */
+export interface MambaModelConfig {
+    vocabSize: number;
+    dModel: number;
+    numLayers: number;
+    dState?: number;
+    dConv?: number;
+    expand?: number;
+    eosId?: number;
+}
+export interface ModelForwardResult {
+    logits: Float32Array;
+    gpuLogits: GPUBuffer;
+    caches: unknown[];
+}
+export interface SamplingOptions {
+    temperature?: number;
+    topK?: number;
+    topP?: number;
+}
+export declare class HybridMambaModel {
+    device: GPUDevice;
+    config: Required<HybridMambaModelConfig>;
+    gpuEmbedding: GPUBuffer;
+    layers: SequenceLayer[];
+    layerSpecs: LayerSpec[];
+    gpuFinalNorm: GPUBuffer;
+    tiedEmbedding: boolean;
+    gpuLMHeadBias: GPUBuffer;
+    private _lmHeadPipeline;
+    private _rmsnormPipeline;
+    private _embedPipeline;
+    private _wslaMode;
+    constructor(device: GPUDevice, config: HybridMambaModelConfig);
+    private _buildLayer;
+    embedTokens(tokenIds: number[] | Uint32Array, batch: number, seqLen: number): GPUBuffer;
+    forward(tokenIds: number[] | Uint32Array, batch: number, seqLen: number): Promise<ModelForwardResult>;
+    /**
+     * Produces a single fixed-length embedding vector for a token sequence.
+     *
+     * Runs the full layer stack plus the final RMSNorm — i.e. the same hidden
+     * state the LM head consumes — then mean-pools across sequence positions and
+     * L2-normalises the result. The returned vector has length `dModel` and is
+     * suitable for cosine-similarity semantic search.
+     *
+     * Unlike `forward()`, this skips the (expensive) LM-head projection: it only
+     * needs the `dModel`-wide hidden state, not `vocabSize` logits.
+     *
+     * The embedding reflects whatever the model currently knows — an untrained
+     * model behaves like a random projection of the token embeddings (still
+     * lexically discriminative), and the representation sharpens automatically as
+     * the model is adapted/distilled.
+     */
+    embed(tokenIds: number[] | Uint32Array): Promise<Float32Array>;
+    generate(promptIds: number[], maxNewTokens?: number, samplingOpts?: SamplingOptions): Promise<number[]>;
+    parameters(): LayerParam[];
+    setWSLAMode(enabled: boolean): void;
+    /**
+     * Export all parameters to an ArrayBuffer.
+     *
+     * MBJS v2/v3 format (identical header; only the data encoding differs):
+     *   [0..3]   magic    : uint32 = 0x4D424A53
+     *   [4..7]   version  : uint32 = 2 (fp32 data) | 3 (fp16 data)
+     *   [8..11]  nLayers  : uint32
+     *   [12 .. 12+nLayers-1]  layerType[i]: uint8 (0=m1, 1=m2, 2=m3, 3=attn)
+     *   aligned to 4 bytes: padding
+     *   [next 4] nParams  : uint32
+     *   [next 4*nParams]  numel[i]: uint32
+     *   [data]  float32 values (v2)  |  float16 values (v3, half the size)
+     *
+     * Pass `{ fp16: true }` to emit a v3 checkpoint — roughly half the bytes,
+     * with a small precision loss that is negligible for SSM weights.
+     */
+    exportWeights(opts?: {
+        fp16?: boolean;
+    }): Promise<ArrayBuffer>;
+    /**
+     * Load parameters from an MBJS v1, v2, or v3 ArrayBuffer.
+     *
+     * v1: assumes all layers are mamba1 (backward compatible).
+     * v2: reads layer type array and validates per-layer parameter counts (fp32 data).
+     * v3: identical layout to v2 but the data section is fp16 (dequantised on load).
+     */
+    loadWeights(buffer: ArrayBuffer): Promise<void>;
+    destroy(): void;
+}
+export declare class MambaModel extends HybridMambaModel {
+    constructor(device: GPUDevice, config: MambaModelConfig);
+}
+//# sourceMappingURL=mamba_model.d.ts.map

package/dist/model/mamba_model.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"mamba_model.d.ts","sourceRoot":"","sources":["../../src/model/mamba_model.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;GAWG;AAMH,OAAO,KAAK,EAAE,aAAa,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,qBAAqB,CAAC;AAChF,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAQ,mBAAmB,CAAC;AAC7D,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAQ,mBAAmB,CAAC;AAC7D,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAQ,mBAAmB,CAAC;AAC7D,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,sBAAsB,CAAC;AAoBjE,MAAM,WAAW,SAAS;IACtB,IAAI,EAAM,SAAS,CAAC;IACpB,MAAM,CAAC,EAAG,OAAO,CAAC,iBAAiB,GAAG,iBAAiB,GAAG,iBAAiB,GAAG,oBAAoB,CAAC,CAAC;CACvG;AAED,MAAM,WAAW,sBAAsB;IACnC,SAAS,EAAU,MAAM,CAAC;IAC1B,MAAM,EAAa,MAAM,CAAC;IAC1B,SAAS,EAAU,MAAM,CAAC;IAE1B;;;OAGG;IACH,MAAM,CAAC,EAAY,SAAS,EAAE,CAAC;IAG/B,aAAa,CAAC,EAAK,OAAO,CAAC,iBAAiB,CAAC,CAAC;IAC9C,aAAa,CAAC,EAAK,OAAO,CAAC,iBAAiB,CAAC,CAAC;IAC9C,aAAa,CAAC,EAAK,OAAO,CAAC,iBAAiB,CAAC,CAAC;IAC9C,gBAAgB,CAAC,EAAE,OAAO,CAAC,oBAAoB,CAAC,CAAC;IAGjD,MAAM,CAAC,EAAY,MAAM,CAAC;IAC1B,KAAK,CAAC,EAAa,MAAM,CAAC;IAC1B,MAAM,CAAC,EAAY,MAAM,CAAC;IAG1B,MAAM,CAAC,EAAY,MAAM,CAAC;IAC1B,OAAO,CAAC,EAAW,MAAM,CAAC;IAC1B,QAAQ,CAAC,EAAU,MAAM,CAAC;IAC1B,SAAS,CAAC,EAAS,MAAM,CAAC;IAE1B,KAAK,CAAC,EAAa,MAAM,CAAC;IAE1B;;;;;OAKG;IACH,IAAI,CAAC,EAAc,MAAM,CAAC;CAC7B;AAED,8DAA8D;AAC9D,MAAM,WAAW,gBAAgB;IAC7B,SAAS,EAAI,MAAM,CAAC;IACpB,MAAM,EAAO,MAAM,CAAC;IACpB,SAAS,EAAI,MAAM,CAAC;IACpB,MAAM,CAAC,EAAM,MAAM,CAAC;IACpB,KAAK,CAAC,EAAO,MAAM,CAAC;IACpB,MAAM,CAAC,EAAM,MAAM,CAAC;IACpB,KAAK,CAAC,EAAO,MAAM,CAAC;CACvB;AAED,MAAM,WAAW,kBAAkB;IAC/B,MAAM,EAAM,YAAY,CAAC;IACzB,SAAS,EAAG,SAAS,CAAC;IACtB,MAAM,EAAM,OAAO,EAAE,CAAC;CACzB;AAED,MAAM,WAAW,eAAe;IAC5B,WAAW,CAAC,EAAG,MAAM,CAAC;IACtB,IAAI,CAAC,EAAU,MAAM,CAAC;IACtB,IAAI,CAAC,EAAU,MAAM,CAAC;CACzB;AAeD,qBAAa,gBAAgB;IACzB,MAAM,EAAU,SAAS,CAAC;IAC1B,MAAM,EAAU,QAAQ,CAAC,sBAAsB,CAAC,CAAC;IACjD,YAAY,EAAI,SAAS,CAAC;IAC1B,MAAM,EAAU,aAAa,EAAE,CAAC;IAChC,UAAU,EAAM,SAAS,EAAE,CAAC;IAC5B,YAAY,EAAI,SAAS,CAAC;IAC1B,aAAa,EAAG,OAAO,CAAC;IACxB,aAAa,EAAG,SAAS,CAAC;IAE1B,OAAO,CAAC,eAAe,CAAuB;IAC9C,OAAO,CAAC,gBAAgB,CAAsB;IAC9C,OAAO,CAAC,cAAc,CAAwB;IAC9C,OAAO,CAAC,SAAS,CAAS;gBAEd,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,sBAAsB;IA0D7D,OAAO,CAAC,WAAW;IAmDnB,WAAW,CAAC,QAAQ,EAAE,MAAM,EAAE,GAAG,WAAW,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,SAAS;IAkBjF,OAAO,CAAC,QAAQ,EAAE,MAAM,EAAE,GAAG,WAAW,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,kBAAkB,CAAC;IA6C3G;;;;;;;;;;;;;;;OAeG;IACG,KAAK,CAAC,QAAQ,EAAE,MAAM,EAAE,GAAG,WAAW,GAAG,OAAO,CAAC,YAAY,CAAC;IAiD9D,QAAQ,CAAC,SAAS,EAAE,MAAM,EAAE,EAAE,YAAY,SAAM,EAAE,YAAY,GAAE,eAAoB,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;IAiB9G,UAAU,IAAI,UAAU,EAAE;IAwB1B,WAAW,CAAC,OAAO,EAAE,OAAO,GAAG,IAAI;IAOnC;;;;;;;;;;;;;;;OAeG;IACG,aAAa,CAAC,IAAI,GAAE;QAAE,IAAI,CAAC,EAAE,OAAO,CAAA;KAAO,GAAG,OAAO,CAAC,WAAW,CAAC;IAqDxE;;;;;;OAMG;IACG,WAAW,CAAC,MAAM,EAAE,WAAW,GAAG,OAAO,CAAC,IAAI,CAAC;IAkGrD,OAAO,IAAI,IAAI;CAMlB;AAID,qBAAa,UAAW,SAAQ,gBAAgB;gBAChC,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,gBAAgB;CAM1D"}