@mni-ml/framework 0.0.1

package/dist/autodiff.d.ts

@@ -0,0 +1,13 @@
+import { Scalar } from "./scalar.js";
+import { Tensor } from "./tensor.js";
+export declare function centralDifference(f: (...args: number[]) => number, vals: number[], arg?: number, epsilon?: number): number;
+export declare class Context {
+    private _savedValues;
+    saveForBackward(...values: number[]): void;
+    get savedValues(): number[];
+}
+export declare function topologicalSort(scalar: Scalar): Scalar[];
+export declare function backPropagate(scalar: Scalar, dOut: number): void;
+export declare function topologicalSortTensor(tensor: Tensor): Tensor[];
+export declare function backPropagateTensor(tensor: Tensor, gradOutput: Tensor): void;
+//# sourceMappingURL=autodiff.d.ts.map

package/dist/autodiff.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"autodiff.d.ts","sourceRoot":"","sources":["../src/autodiff.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AACrC,OAAO,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AAErC,wBAAgB,iBAAiB,CAC7B,CAAC,EAAE,CAAC,GAAG,IAAI,EAAE,MAAM,EAAE,KAAK,MAAM,EAChC,IAAI,EAAE,MAAM,EAAE,EACd,GAAG,GAAE,MAAU,EACf,OAAO,GAAE,MAAa,GACvB,MAAM,CAQR;AAED,qBAAa,OAAO;IAChB,OAAO,CAAC,YAAY,CAAgB;IAEpC,eAAe,CAAE,GAAG,MAAM,EAAE,MAAM,EAAE,GAAG,IAAI;IAI3C,IAAI,WAAW,IAAI,MAAM,EAAE,CAE1B;CACJ;AAED,wBAAgB,eAAe,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,EAAE,CAcxD;AAED,wBAAgB,aAAa,CAAC,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,IAAI,CAkBhE;AAED,wBAAgB,qBAAqB,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,EAAE,CAc9D;AAED,wBAAgB,mBAAmB,CAAC,MAAM,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,GAAG,IAAI,CAuB5E"}

package/dist/autodiff.js

@@ -0,0 +1,91 @@
+import { Scalar } from "./scalar.js";
+import { Tensor } from "./tensor.js";
+export function centralDifference(f, vals, arg = 0, epsilon = 1e-6) {
+    const valsPlus = [...vals];
+    valsPlus[arg] = valsPlus[arg] + epsilon;
+    const valsMinus = [...vals];
+    valsMinus[arg] = valsMinus[arg] - epsilon;
+    return (f(...valsPlus) - f(...valsMinus)) / (2 * epsilon);
+}
+export class Context {
+    _savedValues = [];
+    saveForBackward(...values) {
+        this._savedValues = values;
+    }
+    get savedValues() {
+        return this._savedValues;
+    }
+}
+export function topologicalSort(scalar) {
+    const visited = new Set();
+    const sorted = new Array();
+    const dfs = (scalar) => {
+        if (visited.has(scalar))
+            return;
+        visited.add(scalar);
+        for (const parent of scalar.parents) {
+            dfs(parent);
+        }
+        sorted.push(scalar);
+    };
+    dfs(scalar);
+    return sorted.reverse();
+}
+export function backPropagate(scalar, dOut) {
+    const sorted = topologicalSort(scalar);
+    const derivatives = new Map();
+    derivatives.set(scalar, dOut);
+    for (const node of sorted) {
+        const d = derivatives.get(node);
+        if (d === undefined)
+            continue;
+        if (node.isLeaf()) {
+            node.accumulateDerivative(d);
+        }
+        else {
+            for (const [parent, grad] of node.chainRule(d)) {
+                derivatives.set(parent, (derivatives.get(parent) ?? 0) + grad);
+            }
+        }
+    }
+}
+export function topologicalSortTensor(tensor) {
+    const visited = new Set();
+    const sorted = [];
+    const dfs = (t) => {
+        if (visited.has(t))
+            return;
+        visited.add(t);
+        for (const parent of t.parents) {
+            dfs(parent);
+        }
+        sorted.push(t);
+    };
+    dfs(tensor);
+    return sorted.reverse();
+}
+export function backPropagateTensor(tensor, gradOutput) {
+    const sorted = topologicalSortTensor(tensor);
+    const gradients = new Map();
+    gradients.set(tensor, gradOutput);
+    for (const node of sorted) {
+        const grad = gradients.get(node);
+        if (grad === undefined)
+            continue;
+        if (node.isLeaf()) {
+            node.accumulateGrad(grad);
+        }
+        else {
+            for (const [parent, parentGrad] of node.chainRule(grad)) {
+                const existing = gradients.get(parent);
+                if (existing) {
+                    gradients.set(parent, existing.add(parentGrad));
+                }
+                else {
+                    gradients.set(parent, parentGrad);
+                }
+            }
+        }
+    }
+}
+//# sourceMappingURL=autodiff.js.map

package/dist/autodiff.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"autodiff.js","sourceRoot":"","sources":["../src/autodiff.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AACrC,OAAO,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AAErC,MAAM,UAAU,iBAAiB,CAC7B,CAAgC,EAChC,IAAc,EACd,MAAc,CAAC,EACf,UAAkB,IAAI;IAEtB,MAAM,QAAQ,GAAG,CAAC,GAAG,IAAI,CAAC,CAAC;IAC3B,QAAQ,CAAC,GAAG,CAAC,GAAG,QAAQ,CAAC,GAAG,CAAE,GAAG,OAAO,CAAC;IAEzC,MAAM,SAAS,GAAG,CAAC,GAAG,IAAI,CAAC,CAAC;IAC5B,SAAS,CAAC,GAAG,CAAC,GAAG,SAAS,CAAC,GAAG,CAAE,GAAG,OAAO,CAAC;IAE3C,OAAO,CAAC,CAAC,CAAC,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,GAAG,SAAS,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,OAAO,CAAC,CAAC;AAC9D,CAAC;AAED,MAAM,OAAO,OAAO;IACR,YAAY,GAAa,EAAE,CAAC;IAEpC,eAAe,CAAE,GAAG,MAAgB;QAChC,IAAI,CAAC,YAAY,GAAG,MAAM,CAAC;IAC/B,CAAC;IAED,IAAI,WAAW;QACX,OAAO,IAAI,CAAC,YAAY,CAAC;IAC7B,CAAC;CACJ;AAED,MAAM,UAAU,eAAe,CAAC,MAAc;IAC1C,MAAM,OAAO,GAAG,IAAI,GAAG,EAAU,CAAC;IAClC,MAAM,MAAM,GAAG,IAAI,KAAK,EAAU,CAAC;IAEnC,MAAM,GAAG,GAA6B,CAAC,MAAM,EAAE,EAAE;QAC7C,IAAI,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC;YAAE,OAAO;QAChC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACpB,KAAK,MAAM,MAAM,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;YAClC,GAAG,CAAC,MAAM,CAAC,CAAC;QAChB,CAAC;QACD,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACxB,CAAC,CAAC;IACF,GAAG,CAAC,MAAM,CAAC,CAAC;IACZ,OAAO,MAAM,CAAC,OAAO,EAAE,CAAC;AAC5B,CAAC;AAED,MAAM,UAAU,aAAa,CAAC,MAAc,EAAE,IAAY;IACtD,MAAM,MAAM,GAAG,eAAe,CAAC,MAAM,CAAC,CAAC;IACvC,MAAM,WAAW,GAAwB,IAAI,GAAG,EAAE,CAAC;IAEnD,WAAW,CAAC,GAAG,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;IAE9B,KAAK,MAAM,IAAI,IAAI,MAAM,EAAE,CAAC;QACxB,MAAM,CAAC,GAAG,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAChC,IAAI,CAAC,KAAK,SAAS;YAAE,SAAS;QAE9B,IAAI,IAAI,CAAC,MAAM,EAAE,EAAE,CAAC;YAChB,IAAI,CAAC,oBAAoB,CAAC,CAAC,CAAC,CAAC;QACjC,CAAC;aAAM,CAAC;YACJ,KAAK,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,IAAI,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC;gBAC7C,WAAW,CAAC,GAAG,CAAC,MAAM,EAAE,CAAC,WAAW,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC;YACnE,CAAC;QACL,CAAC;IACL,CAAC;AACL,CAAC;AAED,MAAM,UAAU,qBAAqB,CAAC,MAAc;IAChD,MAAM,OAAO,GAAG,IAAI,GAAG,EAAU,CAAC;IAClC,MAAM,MAAM,GAAa,EAAE,CAAC;IAE5B,MAAM,GAAG,GAAG,CAAC,CAAS,EAAE,EAAE;QACtB,IAAI,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC;YAAE,OAAO;QAC3B,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;QACf,KAAK,MAAM,MAAM,IAAI,CAAC,CAAC,OAAO,EAAE,CAAC;YAC7B,GAAG,CAAC,MAAM,CAAC,CAAC;QAChB,CAAC;QACD,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACnB,CAAC,CAAC;IACF,GAAG,CAAC,MAAM,CAAC,CAAC;IACZ,OAAO,MAAM,CAAC,OAAO,EAAE,CAAC;AAC5B,CAAC;AAED,MAAM,UAAU,mBAAmB,CAAC,MAAc,EAAE,UAAkB;IAClE,MAAM,MAAM,GAAG,qBAAqB,CAAC,MAAM,CAAC,CAAC;IAC7C,MAAM,SAAS,GAAwB,IAAI,GAAG,EAAE,CAAC;IAEjD,SAAS,CAAC,GAAG,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;IAElC,KAAK,MAAM,IAAI,IAAI,MAAM,EAAE,CAAC;QACxB,MAAM,IAAI,GAAG,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QACjC,IAAI,IAAI,KAAK,SAAS;YAAE,SAAS;QAEjC,IAAI,IAAI,CAAC,MAAM,EAAE,EAAE,CAAC;YAChB,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC;QAC9B,CAAC;aAAM,CAAC;YACJ,KAAK,MAAM,CAAC,MAAM,EAAE,UAAU,CAAC,IAAI,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC;gBACtD,MAAM,QAAQ,GAAG,SAAS,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;gBACvC,IAAI,QAAQ,EAAE,CAAC;oBACX,SAAS,CAAC,GAAG,CAAC,MAAM,EAAE,QAAQ,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC;gBACpD,CAAC;qBAAM,CAAC;oBACJ,SAAS,CAAC,GAAG,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;gBACtC,CAAC;YACL,CAAC;QACL,CAAC;IACL,CAAC;AACL,CAAC"}

package/dist/datasets.d.ts

@@ -0,0 +1,16 @@
+export type Point = [number, number];
+export interface Graph {
+    N: number;
+    X: Point[];
+    y: number[];
+}
+/** generate N random 2D points in [0, 1) × [0, 1) */
+export declare function makePts(N: number): Point[];
+export declare function simple(N: number): Graph;
+export declare function diag(N: number): Graph;
+export declare function split(N: number): Graph;
+export declare function xor(N: number): Graph;
+export declare function circle(N: number): Graph;
+export declare function spiral(N: number): Graph;
+export declare const datasets: Record<string, (N: number) => Graph>;
+//# sourceMappingURL=datasets.d.ts.map

package/dist/datasets.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"datasets.d.ts","sourceRoot":"","sources":["../src/datasets.ts"],"names":[],"mappings":"AAAA,MAAM,MAAM,KAAK,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;AAErC,MAAM,WAAW,KAAK;IACpB,CAAC,EAAE,MAAM,CAAC;IACV,CAAC,EAAE,KAAK,EAAE,CAAC;IACX,CAAC,EAAE,MAAM,EAAE,CAAC;CACb;AAED,qDAAqD;AACrD,wBAAgB,OAAO,CAAC,CAAC,EAAE,MAAM,GAAG,KAAK,EAAE,CAQ1C;AAED,wBAAgB,MAAM,CAAC,CAAC,EAAE,MAAM,GAAG,KAAK,CAIvC;AAED,wBAAgB,IAAI,CAAC,CAAC,EAAE,MAAM,GAAG,KAAK,CAIrC;AAED,wBAAgB,KAAK,CAAC,CAAC,EAAE,MAAM,GAAG,KAAK,CAItC;AAED,wBAAgB,GAAG,CAAC,CAAC,EAAE,MAAM,GAAG,KAAK,CAMpC;AAED,wBAAgB,MAAM,CAAC,CAAC,EAAE,MAAM,GAAG,KAAK,CAQvC;AAED,wBAAgB,MAAM,CAAC,CAAC,EAAE,MAAM,GAAG,KAAK,CAoBvC;AAED,eAAO,MAAM,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC,EAAE,MAAM,KAAK,KAAK,CAOzD,CAAC"}

package/dist/datasets.js

@@ -0,0 +1,64 @@
+/** generate N random 2D points in [0, 1) × [0, 1) */
+export function makePts(N) {
+    const X = [];
+    for (let i = 0; i < N; i++) {
+        const x1 = Math.random();
+        const x2 = Math.random();
+        X.push([x1, x2]);
+    }
+    return X;
+}
+export function simple(N) {
+    const X = makePts(N);
+    const y = X.map(([x1]) => (x1 < 0.5 ? 1 : 0));
+    return { N, X, y };
+}
+export function diag(N) {
+    const X = makePts(N);
+    const y = X.map(([x1, x2]) => (x1 + x2 < 0.5 ? 1 : 0));
+    return { N, X, y };
+}
+export function split(N) {
+    const X = makePts(N);
+    const y = X.map(([x1]) => (x1 < 0.2 || x1 > 0.8 ? 1 : 0));
+    return { N, X, y };
+}
+export function xor(N) {
+    const X = makePts(N);
+    const y = X.map(([x1, x2]) => (x1 < 0.5 && x2 > 0.5) || (x1 > 0.5 && x2 < 0.5) ? 1 : 0);
+    return { N, X, y };
+}
+export function circle(N) {
+    const X = makePts(N);
+    const y = X.map(([x1, x2]) => {
+        const dx = x1 - 0.5;
+        const dy = x2 - 0.5;
+        return dx * dx + dy * dy > 0.1 ? 1 : 0;
+    });
+    return { N, X, y };
+}
+export function spiral(N) {
+    const half = Math.floor(N / 2);
+    const fx = (t) => (t * Math.cos(t)) / 20.0;
+    const fy = (t) => (t * Math.sin(t)) / 20.0;
+    const X1 = Array.from({ length: half }, (_, i) => {
+        const t = 10.0 * (i / half);
+        return [fx(t) + 0.5, fy(t) + 0.5];
+    });
+    const X2 = Array.from({ length: half }, (_, i) => {
+        const t = -10.0 * (i / half);
+        return [fy(t) + 0.5, fx(t) + 0.5];
+    });
+    const X = [...X1, ...X2];
+    const y = [...Array(half).fill(0), ...Array(half).fill(1)];
+    return { N, X, y };
+}
+export const datasets = {
+    Simple: simple,
+    Diag: diag,
+    Split: split,
+    Xor: xor,
+    Circle: circle,
+    Spiral: spiral,
+};
+//# sourceMappingURL=datasets.js.map

package/dist/datasets.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"datasets.js","sourceRoot":"","sources":["../src/datasets.ts"],"names":[],"mappings":"AAQA,qDAAqD;AACrD,MAAM,UAAU,OAAO,CAAC,CAAS;IAC/B,MAAM,CAAC,GAAY,EAAE,CAAC;IACtB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QAC3B,MAAM,EAAE,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;QACzB,MAAM,EAAE,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;QACzB,CAAC,CAAC,IAAI,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;IACnB,CAAC;IACD,OAAO,CAAC,CAAC;AACX,CAAC;AAED,MAAM,UAAU,MAAM,CAAC,CAAS;IAC9B,MAAM,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;IACrB,MAAM,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IAC9C,OAAO,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC;AACrB,CAAC;AAED,MAAM,UAAU,IAAI,CAAC,CAAS;IAC5B,MAAM,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;IACrB,MAAM,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,GAAG,EAAE,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IACvD,OAAO,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC;AACrB,CAAC;AAED,MAAM,UAAU,KAAK,CAAC,CAAS;IAC7B,MAAM,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;IACrB,MAAM,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,GAAG,GAAG,IAAI,EAAE,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IAC1D,OAAO,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC;AACrB,CAAC;AAED,MAAM,UAAU,GAAG,CAAC,CAAS;IAC3B,MAAM,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;IACrB,MAAM,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,CAC3B,CAAC,EAAE,GAAG,GAAG,IAAI,EAAE,GAAG,GAAG,CAAC,IAAI,CAAC,EAAE,GAAG,GAAG,IAAI,EAAE,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CACzD,CAAC;IACF,OAAO,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC;AACrB,CAAC;AAED,MAAM,UAAU,MAAM,CAAC,CAAS;IAC9B,MAAM,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;IACrB,MAAM,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE;QAC3B,MAAM,EAAE,GAAG,EAAE,GAAG,GAAG,CAAC;QACpB,MAAM,EAAE,GAAG,EAAE,GAAG,GAAG,CAAC;QACpB,OAAO,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;IACH,OAAO,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC;AACrB,CAAC;AAED,MAAM,UAAU,MAAM,CAAC,CAAS;IAC9B,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;IAE/B,MAAM,EAAE,GAAG,CAAC,CAAS,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC;IACnD,MAAM,EAAE,GAAG,CAAC,CAAS,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC;IAEnD,MAAM,EAAE,GAAY,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE;QACxD,MAAM,CAAC,GAAG,IAAI,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC;QAC5B,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,GAAG,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,GAAG,CAAC,CAAC;IACpC,CAAC,CAAC,CAAC;IAEH,MAAM,EAAE,GAAY,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE;QACxD,MAAM,CAAC,GAAG,CAAC,IAAI,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC;QAC7B,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,GAAG,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,GAAG,CAAC,CAAC;IACpC,CAAC,CAAC,CAAC;IAEH,MAAM,CAAC,GAAG,CAAC,GAAG,EAAE,EAAE,GAAG,EAAE,CAAC,CAAC;IACzB,MAAM,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;IAE3D,OAAO,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC;AACrB,CAAC;AAED,MAAM,CAAC,MAAM,QAAQ,GAAyC;IAC5D,MAAM,EAAE,MAAM;IACd,IAAI,EAAE,IAAI;IACV,KAAK,EAAE,KAAK;IACZ,GAAG,EAAE,GAAG;IACR,MAAM,EAAE,MAAM;IACd,MAAM,EAAE,MAAM;CACf,CAAC"}

package/dist/fast_ops.d.ts

@@ -0,0 +1,23 @@
+import type { Storage, Shape, Strides } from './tensor_data.js';
+/** Terminate the worker pool (call in test teardown so Node can exit). */
+export declare function destroyPool(): void;
+/**
+ * Parallel tensor map. Optimizations: main loop in parallel when
+ * size >= PARALLEL_THRESHOLD; stride-aligned fast path avoids all indexing.
+ *
+ * `fn` must be a pure function (no captured variables) -- workers reconstruct
+ * it from `fn.toString()` via `new Function()`.
+ */
+export declare function fastTensorMap(fn: (x: number) => number): (outStorage: Storage, outShape: Shape, outStrides: Strides, inStorage: Storage, inShape: Shape, inStrides: Strides) => void;
+/**
+ * Parallel tensor zip. Same optimizations and pure-function constraint
+ * as fastTensorMap; stride-aligned when all three tensors match.
+ */
+export declare function fastTensorZip(fn: (a: number, b: number) => number): (outStorage: Storage, outShape: Shape, outStrides: Strides, aStorage: Storage, aShape: Shape, aStrides: Strides, bStorage: Storage, bShape: Shape, bStrides: Strides) => void;
+/**
+ * Parallel tensor reduce. Outer loop (output elements) in parallel; inner
+ * reduction is sequential per element using stride-stepping. Same
+ * pure-function constraint as fastTensorMap.
+ */
+export declare function fastTensorReduce(fn: (acc: number, x: number) => number): (outStorage: Storage, outShape: Shape, outStrides: Strides, aStorage: Storage, aShape: Shape, aStrides: Strides, reduceDim: number) => void;
+//# sourceMappingURL=fast_ops.d.ts.map

package/dist/fast_ops.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"fast_ops.d.ts","sourceRoot":"","sources":["../src/fast_ops.ts"],"names":[],"mappings":"AAKA,OAAO,KAAK,EACR,OAAO,EACP,KAAK,EACL,OAAO,EACV,MAAM,kBAAkB,CAAC;AAgH1B,0EAA0E;AAC1E,wBAAgB,WAAW,IAAI,IAAI,CAKlC;AAMD;;;;;;GAMG;AACH,wBAAgB,aAAa,CACzB,EAAE,EAAE,CAAC,CAAC,EAAE,MAAM,KAAK,MAAM,GAC1B,CACC,UAAU,EAAE,OAAO,EACnB,QAAQ,EAAE,KAAK,EACf,UAAU,EAAE,OAAO,EACnB,SAAS,EAAE,OAAO,EAClB,OAAO,EAAE,KAAK,EACd,SAAS,EAAE,OAAO,KACjB,IAAI,CAyDR;AAED;;;GAGG;AACH,wBAAgB,aAAa,CACzB,EAAE,EAAE,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,MAAM,KAAK,MAAM,GACrC,CACC,UAAU,EAAE,OAAO,EACnB,QAAQ,EAAE,KAAK,EACf,UAAU,EAAE,OAAO,EACnB,QAAQ,EAAE,OAAO,EACjB,MAAM,EAAE,KAAK,EACb,QAAQ,EAAE,OAAO,EACjB,QAAQ,EAAE,OAAO,EACjB,MAAM,EAAE,KAAK,EACb,QAAQ,EAAE,OAAO,KAChB,IAAI,CAsER;AAED;;;;GAIG;AACH,wBAAgB,gBAAgB,CAC5B,EAAE,EAAE,CAAC,GAAG,EAAE,MAAM,EAAE,CAAC,EAAE,MAAM,KAAK,MAAM,GACvC,CACC,UAAU,EAAE,OAAO,EACnB,QAAQ,EAAE,KAAK,EACf,UAAU,EAAE,OAAO,EACnB,QAAQ,EAAE,OAAO,EACjB,MAAM,EAAE,KAAK,EACb,QAAQ,EAAE,OAAO,EACjB,SAAS,EAAE,MAAM,KAChB,IAAI,CA+DR"}

package/dist/fast_ops.js

@@ -0,0 +1,263 @@
+import { Worker } from 'node:worker_threads';
+import { cpus } from 'node:os';
+import { fileURLToPath } from 'node:url';
+import { dirname, join } from 'node:path';
+import { existsSync } from 'node:fs';
+import { indexToPosition, toIndex, shapeProduct, broadcastIndex, } from './tensor_data.js';
+const NUM_WORKERS = Math.max(1, cpus().length);
+// Parallelism is across independent output elements, not within a single
+// reduction, so both paths produce bitwise-identical results for the same fn.
+const PARALLEL_THRESHOLD = 4096;
+function shapesEqual(a, b) {
+    if (a.length !== b.length)
+        return false;
+    for (let i = 0; i < a.length; i++) {
+        if (a[i] !== b[i])
+            return false;
+    }
+    return true;
+}
+function stridesEqual(a, b) {
+    if (a.length !== b.length)
+        return false;
+    for (let i = 0; i < a.length; i++) {
+        if (a[i] !== b[i])
+            return false;
+    }
+    return true;
+}
+class WorkerPool {
+    workers;
+    syncBuffer;
+    syncArray;
+    constructor(numWorkers) {
+        this.syncBuffer = new SharedArrayBuffer(numWorkers * Int32Array.BYTES_PER_ELEMENT);
+        this.syncArray = new Int32Array(this.syncBuffer);
+        // .ts in dev (Node 25+ type-stripping), .js after tsc build
+        const currentDir = dirname(fileURLToPath(import.meta.url));
+        const tsPath = join(currentDir, 'fast_ops_worker.ts');
+        const jsPath = join(currentDir, 'fast_ops_worker.js');
+        const workerPath = existsSync(tsPath) ? tsPath : jsPath;
+        this.workers = Array.from({ length: numWorkers }, (_, id) => {
+            const w = new Worker(workerPath, {
+                workerData: { workerId: id, syncBuffer: this.syncBuffer },
+            });
+            w.on('error', (err) => {
+                console.error(`[fast_ops] Worker ${id} error:`, err);
+            });
+            return w;
+        });
+    }
+    /** Split [0, size) across workers; block until all signal via Atomics. */
+    parallelFor(size, taskFactory) {
+        const numWorkers = this.workers.length;
+        const chunkSize = Math.ceil(size / numWorkers);
+        for (let i = 0; i < numWorkers; i++) {
+            Atomics.store(this.syncArray, i, 0);
+        }
+        for (let i = 0; i < numWorkers; i++) {
+            const start = i * chunkSize;
+            const end = Math.min(start + chunkSize, size);
+            if (start >= size) {
+                Atomics.store(this.syncArray, i, 1);
+                continue;
+            }
+            this.workers[i].postMessage(taskFactory(start, end));
+        }
+        for (let i = 0; i < numWorkers; i++) {
+            while (Atomics.load(this.syncArray, i) === 0) {
+                Atomics.wait(this.syncArray, i, 0, 30000);
+            }
+        }
+    }
+    terminate() {
+        for (const w of this.workers) {
+            w.terminate();
+        }
+    }
+}
+let _pool = undefined;
+// SharedArrayBuffer can't cross Jest's VM context boundary to real workers.
+const PARALLEL_DISABLED = typeof process !== 'undefined' &&
+    (process.env['JEST_WORKER_ID'] !== undefined ||
+        process.env['TSTORCH_DISABLE_PARALLEL'] !== undefined);
+function getPool() {
+    if (PARALLEL_DISABLED)
+        return null;
+    if (_pool === undefined) {
+        try {
+            _pool = new WorkerPool(NUM_WORKERS);
+        }
+        catch {
+            _pool = null;
+        }
+    }
+    return _pool;
+}
+/** Terminate the worker pool (call in test teardown so Node can exit). */
+export function destroyPool() {
+    if (_pool) {
+        _pool.terminate();
+        _pool = undefined;
+    }
+}
+function isShared(storage) {
+    return storage.buffer instanceof SharedArrayBuffer;
+}
+/**
+ * Parallel tensor map. Optimizations: main loop in parallel when
+ * size >= PARALLEL_THRESHOLD; stride-aligned fast path avoids all indexing.
+ *
+ * `fn` must be a pure function (no captured variables) -- workers reconstruct
+ * it from `fn.toString()` via `new Function()`.
+ */
+export function fastTensorMap(fn) {
+    return (outStorage, outShape, outStrides, inStorage, inShape, inStrides) => {
+        const size = shapeProduct(outShape);
+        const aligned = shapesEqual(outShape, inShape) && stridesEqual(outStrides, inStrides);
+        const pool = getPool();
+        if (pool &&
+            size >= PARALLEL_THRESHOLD &&
+            isShared(outStorage) &&
+            isShared(inStorage)) {
+            const fnSource = fn.toString();
+            pool.parallelFor(size, (start, end) => ({
+                type: 'map',
+                fnSource,
+                start,
+                end,
+                outBuffer: outStorage.buffer,
+                outShape: Array.from(outShape),
+                outStrides: Array.from(outStrides),
+                inBuffer: inStorage.buffer,
+                inShape: Array.from(inShape),
+                inStrides: Array.from(inStrides),
+                aligned,
+            }));
+            return;
+        }
+        if (aligned) {
+            for (let i = 0; i < size; i++) {
+                outStorage[i] = fn(inStorage[i]);
+            }
+            return;
+        }
+        const outIndex = new Array(outShape.length).fill(0);
+        const inIndex = new Array(inShape.length).fill(0);
+        for (let ordinal = 0; ordinal < size; ordinal++) {
+            toIndex(ordinal, outShape, outIndex);
+            broadcastIndex(outIndex, outShape, inShape, inIndex);
+            const inPos = indexToPosition(inIndex, inStrides);
+            const outPos = indexToPosition(outIndex, outStrides);
+            outStorage[outPos] = fn(inStorage[inPos]);
+        }
+    };
+}
+/**
+ * Parallel tensor zip. Same optimizations and pure-function constraint
+ * as fastTensorMap; stride-aligned when all three tensors match.
+ */
+export function fastTensorZip(fn) {
+    return (outStorage, outShape, outStrides, aStorage, aShape, aStrides, bStorage, bShape, bStrides) => {
+        const size = shapeProduct(outShape);
+        const aligned = shapesEqual(outShape, aShape) &&
+            shapesEqual(outShape, bShape) &&
+            stridesEqual(outStrides, aStrides) &&
+            stridesEqual(outStrides, bStrides);
+        const pool = getPool();
+        if (pool &&
+            size >= PARALLEL_THRESHOLD &&
+            isShared(outStorage) &&
+            isShared(aStorage) &&
+            isShared(bStorage)) {
+            const fnSource = fn.toString();
+            pool.parallelFor(size, (start, end) => ({
+                type: 'zip',
+                fnSource,
+                start,
+                end,
+                outBuffer: outStorage.buffer,
+                outShape: Array.from(outShape),
+                outStrides: Array.from(outStrides),
+                aBuffer: aStorage.buffer,
+                aShape: Array.from(aShape),
+                aStrides: Array.from(aStrides),
+                bBuffer: bStorage.buffer,
+                bShape: Array.from(bShape),
+                bStrides: Array.from(bStrides),
+                aligned,
+            }));
+            return;
+        }
+        if (aligned) {
+            for (let i = 0; i < size; i++) {
+                outStorage[i] = fn(aStorage[i], bStorage[i]);
+            }
+            return;
+        }
+        const outIndex = new Array(outShape.length).fill(0);
+        const aIndex = new Array(aShape.length).fill(0);
+        const bIndex = new Array(bShape.length).fill(0);
+        for (let ordinal = 0; ordinal < size; ordinal++) {
+            toIndex(ordinal, outShape, outIndex);
+            broadcastIndex(outIndex, outShape, aShape, aIndex);
+            broadcastIndex(outIndex, outShape, bShape, bIndex);
+            const aPos = indexToPosition(aIndex, aStrides);
+            const bPos = indexToPosition(bIndex, bStrides);
+            const outPos = indexToPosition(outIndex, outStrides);
+            outStorage[outPos] = fn(aStorage[aPos], bStorage[bPos]);
+        }
+    };
+}
+/**
+ * Parallel tensor reduce. Outer loop (output elements) in parallel; inner
+ * reduction is sequential per element using stride-stepping. Same
+ * pure-function constraint as fastTensorMap.
+ */
+export function fastTensorReduce(fn) {
+    return (outStorage, outShape, outStrides, aStorage, aShape, aStrides, reduceDim) => {
+        const outSize = shapeProduct(outShape);
+        const reduceDimSize = aShape[reduceDim];
+        const reduceStride = aStrides[reduceDim];
+        const pool = getPool();
+        if (pool &&
+            outSize >= PARALLEL_THRESHOLD &&
+            isShared(outStorage) &&
+            isShared(aStorage)) {
+            const fnSource = fn.toString();
+            pool.parallelFor(outSize, (start, end) => ({
+                type: 'reduce',
+                fnSource,
+                start,
+                end,
+                outBuffer: outStorage.buffer,
+                outShape: Array.from(outShape),
+                outStrides: Array.from(outStrides),
+                inBuffer: aStorage.buffer,
+                inShape: Array.from(aShape),
+                inStrides: Array.from(aStrides),
+                reduceDim,
+                reduceDimSize,
+            }));
+            return;
+        }
+        const outIndex = new Array(outShape.length).fill(0);
+        const aIndex = new Array(aShape.length).fill(0);
+        for (let ordinal = 0; ordinal < outSize; ordinal++) {
+            toIndex(ordinal, outShape, outIndex);
+            const outPos = indexToPosition(outIndex, outStrides);
+            for (let i = 0; i < outShape.length; i++) {
+                aIndex[i] = outIndex[i];
+            }
+            aIndex[reduceDim] = 0;
+            let aPos = indexToPosition(aIndex, aStrides);
+            let acc = aStorage[aPos];
+            for (let j = 1; j < reduceDimSize; j++) {
+                aPos += reduceStride;
+                acc = fn(acc, aStorage[aPos]);
+            }
+            outStorage[outPos] = acc;
+        }
+    };
+}
+//# sourceMappingURL=fast_ops.js.map

package/dist/fast_ops.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"fast_ops.js","sourceRoot":"","sources":["../src/fast_ops.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,qBAAqB,CAAC;AAC7C,OAAO,EAAE,IAAI,EAAE,MAAM,SAAS,CAAC;AAC/B,OAAO,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC;AACzC,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AAC1C,OAAO,EAAE,UAAU,EAAE,MAAM,SAAS,CAAC;AAOrC,OAAO,EACH,eAAe,EACf,OAAO,EACP,YAAY,EACZ,cAAc,GACjB,MAAM,kBAAkB,CAAC;AAE1B,MAAM,WAAW,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,MAAM,CAAC,CAAC;AAE/C,yEAAyE;AACzE,8EAA8E;AAC9E,MAAM,kBAAkB,GAAG,IAAI,CAAC;AAEhC,SAAS,WAAW,CAAC,CAAQ,EAAE,CAAQ;IACnC,IAAI,CAAC,CAAC,MAAM,KAAK,CAAC,CAAC,MAAM;QAAE,OAAO,KAAK,CAAC;IACxC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QAChC,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;YAAE,OAAO,KAAK,CAAC;IACpC,CAAC;IACD,OAAO,IAAI,CAAC;AAChB,CAAC;AAED,SAAS,YAAY,CAAC,CAAU,EAAE,CAAU;IACxC,IAAI,CAAC,CAAC,MAAM,KAAK,CAAC,CAAC,MAAM;QAAE,OAAO,KAAK,CAAC;IACxC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QAChC,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;YAAE,OAAO,KAAK,CAAC;IACpC,CAAC;IACD,OAAO,IAAI,CAAC;AAChB,CAAC;AAED,MAAM,UAAU;IACJ,OAAO,CAAW;IAClB,UAAU,CAAoB;IAC9B,SAAS,CAAa;IAE9B,YAAY,UAAkB;QAC1B,IAAI,CAAC,UAAU,GAAG,IAAI,iBAAiB,CAAC,UAAU,GAAG,UAAU,CAAC,iBAAiB,CAAC,CAAC;QACnF,IAAI,CAAC,SAAS,GAAG,IAAI,UAAU,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAEjD,4DAA4D;QAC5D,MAAM,UAAU,GAAG,OAAO,CAAC,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;QAC3D,MAAM,MAAM,GAAG,IAAI,CAAC,UAAU,EAAE,oBAAoB,CAAC,CAAC;QACtD,MAAM,MAAM,GAAG,IAAI,CAAC,UAAU,EAAE,oBAAoB,CAAC,CAAC;QACtD,MAAM,UAAU,GAAG,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC;QAExD,IAAI,CAAC,OAAO,GAAG,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,UAAU,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,EAAE;YACxD,MAAM,CAAC,GAAG,IAAI,MAAM,CAAC,UAAU,EAAE;gBAC7B,UAAU,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE,UAAU,EAAE,IAAI,CAAC,UAAU,EAAE;aAC5D,CAAC,CAAC;YACH,CAAC,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAAG,EAAE,EAAE;gBAClB,OAAO,CAAC,KAAK,CAAC,qBAAqB,EAAE,SAAS,EAAE,GAAG,CAAC,CAAC;YACzD,CAAC,CAAC,CAAC;YACH,OAAO,CAAC,CAAC;QACb,CAAC,CAAC,CAAC;IACP,CAAC;IAED,0EAA0E;IAC1E,WAAW,CACP,IAAY,EACZ,WAAmD;QAEnD,MAAM,UAAU,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC;QACvC,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,GAAG,UAAU,CAAC,CAAC;QAE/C,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;YAClC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;QACxC,CAAC;QAED,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;YAClC,MAAM,KAAK,GAAG,CAAC,GAAG,SAAS,CAAC;YAC5B,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,KAAK,GAAG,SAAS,EAAE,IAAI,CAAC,CAAC;YAC9C,IAAI,KAAK,IAAI,IAAI,EAAE,CAAC;gBAChB,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;gBACpC,SAAS;YACb,CAAC;YACD,IAAI,CAAC,OAAO,CAAC,CAAC,CAAE,CAAC,WAAW,CAAC,WAAW,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC,CAAC;QAC1D,CAAC;QAED,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;YAClC,OAAO,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC,KAAK,CAAC,EAAE,CAAC;gBAC3C,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC,EAAE,CAAC,EAAE,KAAK,CAAC,CAAC;YAC9C,CAAC;QACL,CAAC;IACL,CAAC;IAED,SAAS;QACL,KAAK,MAAM,CAAC,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YAC3B,CAAC,CAAC,SAAS,EAAE,CAAC;QAClB,CAAC;IACL,CAAC;CACJ;AAED,IAAI,KAAK,GAAkC,SAAS,CAAC;AAErD,4EAA4E;AAC5E,MAAM,iBAAiB,GAAG,OAAO,OAAO,KAAK,WAAW;IACpD,CAAC,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC,KAAK,SAAS;QAC3C,OAAO,CAAC,GAAG,CAAC,0BAA0B,CAAC,KAAK,SAAS,CAAC,CAAC;AAE5D,SAAS,OAAO;IACZ,IAAI,iBAAiB;QAAE,OAAO,IAAI,CAAC;IACnC,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;QACtB,IAAI,CAAC;YACD,KAAK,GAAG,IAAI,UAAU,CAAC,WAAW,CAAC,CAAC;QACxC,CAAC;QAAC,MAAM,CAAC;YACL,KAAK,GAAG,IAAI,CAAC;QACjB,CAAC;IACL,CAAC;IACD,OAAO,KAAK,CAAC;AACjB,CAAC;AAED,0EAA0E;AAC1E,MAAM,UAAU,WAAW;IACvB,IAAI,KAAK,EAAE,CAAC;QACR,KAAK,CAAC,SAAS,EAAE,CAAC;QAClB,KAAK,GAAG,SAAS,CAAC;IACtB,CAAC;AACL,CAAC;AAED,SAAS,QAAQ,CAAC,OAAgB;IAC9B,OAAO,OAAO,CAAC,MAAM,YAAY,iBAAiB,CAAC;AACvD,CAAC;AAED;;;;;;GAMG;AACH,MAAM,UAAU,aAAa,CACzB,EAAyB;IASzB,OAAO,CACH,UAAmB,EACnB,QAAe,EACf,UAAmB,EACnB,SAAkB,EAClB,OAAc,EACd,SAAkB,EACd,EAAE;QACN,MAAM,IAAI,GAAG,YAAY,CAAC,QAAQ,CAAC,CAAC;QACpC,MAAM,OAAO,GACT,WAAW,CAAC,QAAQ,EAAE,OAAO,CAAC,IAAI,YAAY,CAAC,UAAU,EAAE,SAAS,CAAC,CAAC;QAE1E,MAAM,IAAI,GAAG,OAAO,EAAE,CAAC;QACvB,IACI,IAAI;YACJ,IAAI,IAAI,kBAAkB;YAC1B,QAAQ,CAAC,UAAU,CAAC;YACpB,QAAQ,CAAC,SAAS,CAAC,EACrB,CAAC;YACC,MAAM,QAAQ,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAC;YAE/B,IAAI,CAAC,WAAW,CAAC,IAAI,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;gBACpC,IAAI,EAAE,KAAK;gBACX,QAAQ;gBACR,KAAK;gBACL,GAAG;gBACH,SAAS,EAAE,UAAU,CAAC,MAA2B;gBACjD,QAAQ,EAAE,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC;gBAC9B,UAAU,EAAE,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC;gBAClC,QAAQ,EAAE,SAAS,CAAC,MAA2B;gBAC/C,OAAO,EAAE,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC;gBAC5B,SAAS,EAAE,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC;gBAChC,OAAO;aACV,CAAC,CAAC,CAAC;YACJ,OAAO;QACX,CAAC;QAED,IAAI,OAAO,EAAE,CAAC;YACV,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC5B,UAAU,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC,CAAE,CAAC,CAAC;YACtC,CAAC;YACD,OAAO;QACX,CAAC;QAED,MAAM,QAAQ,GAAa,IAAI,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC9D,MAAM,OAAO,GAAa,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAE5D,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,GAAG,IAAI,EAAE,OAAO,EAAE,EAAE,CAAC;YAC9C,OAAO,CAAC,OAAO,EAAE,QAAQ,EAAE,QAAQ,CAAC,CAAC;YACrC,cAAc,CAAC,QAAQ,EAAE,QAAQ,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;YAErD,MAAM,KAAK,GAAG,eAAe,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;YAClD,MAAM,MAAM,GAAG,eAAe,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;YACrD,UAAU,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,KAAK,CAAE,CAAC,CAAC;QAC/C,CAAC;IACL,CAAC,CAAC;AACN,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,aAAa,CACzB,EAAoC;IAYpC,OAAO,CACH,UAAmB,EACnB,QAAe,EACf,UAAmB,EACnB,QAAiB,EACjB,MAAa,EACb,QAAiB,EACjB,QAAiB,EACjB,MAAa,EACb,QAAiB,EACb,EAAE;QACN,MAAM,IAAI,GAAG,YAAY,CAAC,QAAQ,CAAC,CAAC;QACpC,MAAM,OAAO,GACT,WAAW,CAAC,QAAQ,EAAE,MAAM,CAAC;YAC7B,WAAW,CAAC,QAAQ,EAAE,MAAM,CAAC;YAC7B,YAAY,CAAC,UAAU,EAAE,QAAQ,CAAC;YAClC,YAAY,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;QAEvC,MAAM,IAAI,GAAG,OAAO,EAAE,CAAC;QACvB,IACI,IAAI;YACJ,IAAI,IAAI,kBAAkB;YAC1B,QAAQ,CAAC,UAAU,CAAC;YACpB,QAAQ,CAAC,QAAQ,CAAC;YAClB,QAAQ,CAAC,QAAQ,CAAC,EACpB,CAAC;YACC,MAAM,QAAQ,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAC;YAE/B,IAAI,CAAC,WAAW,CAAC,IAAI,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;gBACpC,IAAI,EAAE,KAAK;gBACX,QAAQ;gBACR,KAAK;gBACL,GAAG;gBACH,SAAS,EAAE,UAAU,CAAC,MAA2B;gBACjD,QAAQ,EAAE,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC;gBAC9B,UAAU,EAAE,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC;gBAClC,OAAO,EAAE,QAAQ,CAAC,MAA2B;gBAC7C,MAAM,EAAE,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC;gBAC1B,QAAQ,EAAE,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC;gBAC9B,OAAO,EAAE,QAAQ,CAAC,MAA2B;gBAC7C,MAAM,EAAE,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC;gBAC1B,QAAQ,EAAE,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC;gBAC9B,OAAO;aACV,CAAC,CAAC,CAAC;YACJ,OAAO;QACX,CAAC;QAED,IAAI,OAAO,EAAE,CAAC;YACV,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC5B,UAAU,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,CAAC,CAAE,EAAE,QAAQ,CAAC,CAAC,CAAE,CAAC,CAAC;YACnD,CAAC;YACD,OAAO;QACX,CAAC;QAED,MAAM,QAAQ,GAAa,IAAI,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC9D,MAAM,MAAM,GAAa,IAAI,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC1D,MAAM,MAAM,GAAa,IAAI,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAE1D,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,GAAG,IAAI,EAAE,OAAO,EAAE,EAAE,CAAC;YAC9C,OAAO,CAAC,OAAO,EAAE,QAAQ,EAAE,QAAQ,CAAC,CAAC;YACrC,cAAc,CAAC,QAAQ,EAAE,QAAQ,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC;YACnD,cAAc,CAAC,QAAQ,EAAE,QAAQ,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC;YAEnD,MAAM,IAAI,GAAG,eAAe,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;YAC/C,MAAM,IAAI,GAAG,eAAe,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;YAC/C,MAAM,MAAM,GAAG,eAAe,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;YACrD,UAAU,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAE,EAAE,QAAQ,CAAC,IAAI,CAAE,CAAC,CAAC;QAC9D,CAAC;IACL,CAAC,CAAC;AACN,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,gBAAgB,CAC5B,EAAsC;IAUtC,OAAO,CACH,UAAmB,EACnB,QAAe,EACf,UAAmB,EACnB,QAAiB,EACjB,MAAa,EACb,QAAiB,EACjB,SAAiB,EACb,EAAE;QACN,MAAM,OAAO,GAAG,YAAY,CAAC,QAAQ,CAAC,CAAC;QACvC,MAAM,aAAa,GAAG,MAAM,CAAC,SAAS,CAAE,CAAC;QACzC,MAAM,YAAY,GAAG,QAAQ,CAAC,SAAS,CAAE,CAAC;QAE1C,MAAM,IAAI,GAAG,OAAO,EAAE,CAAC;QACvB,IACI,IAAI;YACJ,OAAO,IAAI,kBAAkB;YAC7B,QAAQ,CAAC,UAAU,CAAC;YACpB,QAAQ,CAAC,QAAQ,CAAC,EACpB,CAAC;YACC,MAAM,QAAQ,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAC;YAE/B,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;gBACvC,IAAI,EAAE,QAAQ;gBACd,QAAQ;gBACR,KAAK;gBACL,GAAG;gBACH,SAAS,EAAE,UAAU,CAAC,MAA2B;gBACjD,QAAQ,EAAE,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC;gBAC9B,UAAU,EAAE,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC;gBAClC,QAAQ,EAAE,QAAQ,CAAC,MAA2B;gBAC9C,OAAO,EAAE,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC;gBAC3B,SAAS,EAAE,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC;gBAC/B,SAAS;gBACT,aAAa;aAChB,CAAC,CAAC,CAAC;YACJ,OAAO;QACX,CAAC;QAED,MAAM,QAAQ,GAAa,IAAI,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC9D,MAAM,MAAM,GAAa,IAAI,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAE1D,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,GAAG,OAAO,EAAE,OAAO,EAAE,EAAE,CAAC;YACjD,OAAO,CAAC,OAAO,EAAE,QAAQ,EAAE,QAAQ,CAAC,CAAC;YAErC,MAAM,MAAM,GAAG,eAAe,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;YAErD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,QAAQ,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;gBACvC,MAAM,CAAC,CAAC,CAAC,GAAG,QAAQ,CAAC,CAAC,CAAE,CAAC;YAC7B,CAAC;YACD,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;YACtB,IAAI,IAAI,GAAG,eAAe,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;YAE7C,IAAI,GAAG,GAAG,QAAQ,CAAC,IAAI,CAAE,CAAC;YAC1B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,aAAa,EAAE,CAAC,EAAE,EAAE,CAAC;gBACrC,IAAI,IAAI,YAAY,CAAC;gBACrB,GAAG,GAAG,EAAE,CAAC,GAAG,EAAE,QAAQ,CAAC,IAAI,CAAE,CAAC,CAAC;YACnC,CAAC;YAED,UAAU,CAAC,MAAM,CAAC,GAAG,GAAG,CAAC;QAC7B,CAAC;IACL,CAAC,CAAC;AACN,CAAC"}

package/dist/fast_ops_worker.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=fast_ops_worker.d.ts.map

package/dist/fast_ops_worker.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"fast_ops_worker.d.ts","sourceRoot":"","sources":["../src/fast_ops_worker.ts"],"names":[],"mappings":""}

package/dist/fast_ops_worker.js

@@ -0,0 +1,119 @@
+import { parentPort, workerData } from 'node:worker_threads';
+// Inlined from tensor_data.ts -- workers are separate V8 isolates and
+// can't import .js-extensioned modules from .ts source.
+function toIndex(ordinal, shape, outIndex) {
+    let remaining = ordinal;
+    for (let i = shape.length - 1; i >= 0; i--) {
+        const dimSize = shape[i];
+        outIndex[i] = remaining % dimSize;
+        remaining = Math.floor(remaining / dimSize);
+    }
+}
+function indexToPosition(idx, strides) {
+    let position = 0;
+    for (let i = 0; i < idx.length; i++) {
+        position += idx[i] * strides[i];
+    }
+    return position;
+}
+function broadcastIndex(bigIndex, bigShape, shape, outIndex) {
+    const offset = bigShape.length - shape.length;
+    for (let i = 0; i < shape.length; i++) {
+        const bigI = i + offset;
+        if (shape[i] === 1) {
+            outIndex[i] = 0;
+        }
+        else {
+            outIndex[i] = bigIndex[bigI];
+        }
+    }
+}
+const { workerId, syncBuffer } = workerData;
+const syncArray = new Int32Array(syncBuffer);
+// fn must be pure (no closures) -- reconstructed from source via new Function()
+function reconstructFn(source) {
+    return new Function('return ' + source)();
+}
+function handleMap(task) {
+    const fn = reconstructFn(task.fnSource);
+    const outStorage = new Float64Array(task.outBuffer);
+    const inStorage = new Float64Array(task.inBuffer);
+    if (task.aligned) {
+        for (let i = task.start; i < task.end; i++) {
+            outStorage[i] = fn(inStorage[i]);
+        }
+    }
+    else {
+        const outIndex = new Array(task.outShape.length).fill(0);
+        const inIndex = new Array(task.inShape.length).fill(0);
+        for (let ordinal = task.start; ordinal < task.end; ordinal++) {
+            toIndex(ordinal, task.outShape, outIndex);
+            broadcastIndex(outIndex, task.outShape, task.inShape, inIndex);
+            outStorage[indexToPosition(outIndex, task.outStrides)] =
+                fn(inStorage[indexToPosition(inIndex, task.inStrides)]);
+        }
+    }
+}
+function handleZip(task) {
+    const fn = reconstructFn(task.fnSource);
+    const outStorage = new Float64Array(task.outBuffer);
+    const aStorage = new Float64Array(task.aBuffer);
+    const bStorage = new Float64Array(task.bBuffer);
+    if (task.aligned) {
+        for (let i = task.start; i < task.end; i++) {
+            outStorage[i] = fn(aStorage[i], bStorage[i]);
+        }
+    }
+    else {
+        const outIndex = new Array(task.outShape.length).fill(0);
+        const aIndex = new Array(task.aShape.length).fill(0);
+        const bIndex = new Array(task.bShape.length).fill(0);
+        for (let ordinal = task.start; ordinal < task.end; ordinal++) {
+            toIndex(ordinal, task.outShape, outIndex);
+            broadcastIndex(outIndex, task.outShape, task.aShape, aIndex);
+            broadcastIndex(outIndex, task.outShape, task.bShape, bIndex);
+            outStorage[indexToPosition(outIndex, task.outStrides)] =
+                fn(aStorage[indexToPosition(aIndex, task.aStrides)], bStorage[indexToPosition(bIndex, task.bStrides)]);
+        }
+    }
+}
+function handleReduce(task) {
+    const fn = reconstructFn(task.fnSource);
+    const outStorage = new Float64Array(task.outBuffer);
+    const inStorage = new Float64Array(task.inBuffer);
+    const reduceStride = task.inStrides[task.reduceDim];
+    const outIndex = new Array(task.outShape.length).fill(0);
+    const inIndex = new Array(task.inShape.length).fill(0);
+    for (let ordinal = task.start; ordinal < task.end; ordinal++) {
+        toIndex(ordinal, task.outShape, outIndex);
+        const outPos = indexToPosition(outIndex, task.outStrides);
+        for (let i = 0; i < task.outShape.length; i++) {
+            inIndex[i] = outIndex[i];
+        }
+        inIndex[task.reduceDim] = 0;
+        let inPos = indexToPosition(inIndex, task.inStrides);
+        let acc = inStorage[inPos];
+        for (let j = 1; j < task.reduceDimSize; j++) {
+            inPos += reduceStride;
+            acc = fn(acc, inStorage[inPos]);
+        }
+        outStorage[outPos] = acc;
+    }
+}
+parentPort.on('message', (task) => {
+    switch (task.type) {
+        case 'map':
+            handleMap(task);
+            break;
+        case 'zip':
+            handleZip(task);
+            break;
+        case 'reduce':
+            handleReduce(task);
+            break;
+    }
+    Atomics.store(syncArray, workerId, 1);
+    Atomics.notify(syncArray, workerId);
+    parentPort.postMessage('done');
+});
+//# sourceMappingURL=fast_ops_worker.js.map

package/dist/fast_ops_worker.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"fast_ops_worker.js","sourceRoot":"","sources":["../src/fast_ops_worker.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,qBAAqB,CAAC;AAE7D,sEAAsE;AACtE,wDAAwD;AAExD,SAAS,OAAO,CAAC,OAAe,EAAE,KAAe,EAAE,QAAkB;IACjE,IAAI,SAAS,GAAG,OAAO,CAAC;IACxB,KAAK,IAAI,CAAC,GAAG,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QACzC,MAAM,OAAO,GAAG,KAAK,CAAC,CAAC,CAAE,CAAC;QAC1B,QAAQ,CAAC,CAAC,CAAC,GAAG,SAAS,GAAG,OAAO,CAAC;QAClC,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,GAAG,OAAO,CAAC,CAAC;IAChD,CAAC;AACL,CAAC;AAED,SAAS,eAAe,CAAC,GAAa,EAAE,OAAiB;IACrD,IAAI,QAAQ,GAAG,CAAC,CAAC;IACjB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,GAAG,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QAClC,QAAQ,IAAI,GAAG,CAAC,CAAC,CAAE,GAAG,OAAO,CAAC,CAAC,CAAE,CAAC;IACtC,CAAC;IACD,OAAO,QAAQ,CAAC;AACpB,CAAC;AAED,SAAS,cAAc,CACnB,QAAkB,EAClB,QAAkB,EAClB,KAAe,EACf,QAAkB;IAElB,MAAM,MAAM,GAAG,QAAQ,CAAC,MAAM,GAAG,KAAK,CAAC,MAAM,CAAC;IAC9C,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACpC,MAAM,IAAI,GAAG,CAAC,GAAG,MAAM,CAAC;QACxB,IAAI,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,EAAE,CAAC;YACjB,QAAQ,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;QACpB,CAAC;aAAM,CAAC;YACJ,QAAQ,CAAC,CAAC,CAAC,GAAG,QAAQ,CAAC,IAAI,CAAE,CAAC;QAClC,CAAC;IACL,CAAC;AACL,CAAC;AAkDD,MAAM,EAAE,QAAQ,EAAE,UAAU,EAAE,GAAG,UAGhC,CAAC;AACF,MAAM,SAAS,GAAG,IAAI,UAAU,CAAC,UAAU,CAAC,CAAC;AAE7C,gFAAgF;AAChF,SAAS,aAAa,CAAI,MAAc;IACpC,OAAO,IAAI,QAAQ,CAAC,SAAS,GAAG,MAAM,CAAC,EAAO,CAAC;AACnD,CAAC;AAED,SAAS,SAAS,CAAC,IAAa;IAC5B,MAAM,EAAE,GAAG,aAAa,CAAwB,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC/D,MAAM,UAAU,GAAG,IAAI,YAAY,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IACpD,MAAM,SAAS,GAAG,IAAI,YAAY,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAElD,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;QACf,KAAK,IAAI,CAAC,GAAG,IAAI,CAAC,KAAK,EAAE,CAAC,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YACzC,UAAU,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC,CAAE,CAAC,CAAC;QACtC,CAAC;IACL,CAAC;SAAM,CAAC;QACJ,MAAM,QAAQ,GAAa,IAAI,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACnE,MAAM,OAAO,GAAa,IAAI,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAEjE,KAAK,IAAI,OAAO,GAAG,IAAI,CAAC,KAAK,EAAE,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,OAAO,EAAE,EAAE,CAAC;YAC3D,OAAO,CAAC,OAAO,EAAE,IAAI,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;YAC1C,cAAc,CAAC,QAAQ,EAAE,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YAC/D,UAAU,CAAC,eAAe,CAAC,QAAQ,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC;gBAClD,EAAE,CAAC,SAAS,CAAC,eAAe,CAAC,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,CAAE,CAAC,CAAC;QACjE,CAAC;IACL,CAAC;AACL,CAAC;AAED,SAAS,SAAS,CAAC,IAAa;IAC5B,MAAM,EAAE,GAAG,aAAa,CAAmC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC1E,MAAM,UAAU,GAAG,IAAI,YAAY,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IACpD,MAAM,QAAQ,GAAG,IAAI,YAAY,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAChD,MAAM,QAAQ,GAAG,IAAI,YAAY,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAEhD,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;QACf,KAAK,IAAI,CAAC,GAAG,IAAI,CAAC,KAAK,EAAE,CAAC,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YACzC,UAAU,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,CAAC,CAAE,EAAE,QAAQ,CAAC,CAAC,CAAE,CAAC,CAAC;QACnD,CAAC;IACL,CAAC;SAAM,CAAC;QACJ,MAAM,QAAQ,GAAa,IAAI,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACnE,MAAM,MAAM,GAAa,IAAI,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC/D,MAAM,MAAM,GAAa,IAAI,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAE/D,KAAK,IAAI,OAAO,GAAG,IAAI,CAAC,KAAK,EAAE,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,OAAO,EAAE,EAAE,CAAC;YAC3D,OAAO,CAAC,OAAO,EAAE,IAAI,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;YAC1C,cAAc,CAAC,QAAQ,EAAE,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;YAC7D,cAAc,CAAC,QAAQ,EAAE,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;YAC7D,UAAU,CAAC,eAAe,CAAC,QAAQ,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC;gBAClD,EAAE,CAAC,QAAQ,CAAC,eAAe,CAAC,MAAM,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAE,EACjD,QAAQ,CAAC,eAAe,CAAC,MAAM,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAE,CAAC,CAAC;QAC9D,CAAC;IACL,CAAC;AACL,CAAC;AAED,SAAS,YAAY,CAAC,IAAgB;IAClC,MAAM,EAAE,GAAG,aAAa,CAAqC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC5E,MAAM,UAAU,GAAG,IAAI,YAAY,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IACpD,MAAM,SAAS,GAAG,IAAI,YAAY,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAClD,MAAM,YAAY,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,SAAS,CAAE,CAAC;IAErD,MAAM,QAAQ,GAAa,IAAI,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACnE,MAAM,OAAO,GAAa,IAAI,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAEjE,KAAK,IAAI,OAAO,GAAG,IAAI,CAAC,KAAK,EAAE,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,OAAO,EAAE,EAAE,CAAC;QAC3D,OAAO,CAAC,OAAO,EAAE,IAAI,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;QAC1C,MAAM,MAAM,GAAG,eAAe,CAAC,QAAQ,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC;QAE1D,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YAC5C,OAAO,CAAC,CAAC,CAAC,GAAG,QAAQ,CAAC,CAAC,CAAE,CAAC;QAC9B,CAAC;QACD,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;QAC5B,IAAI,KAAK,GAAG,eAAe,CAAC,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;QAErD,IAAI,GAAG,GAAG,SAAS,CAAC,KAAK,CAAE,CAAC;QAC5B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,aAAa,EAAE,CAAC,EAAE,EAAE,CAAC;YAC1C,KAAK,IAAI,YAAY,CAAC;YACtB,GAAG,GAAG,EAAE,CAAC,GAAG,EAAE,SAAS,CAAC,KAAK,CAAE,CAAC,CAAC;QACrC,CAAC;QAED,UAAU,CAAC,MAAM,CAAC,GAAG,GAAG,CAAC;IAC7B,CAAC;AACL,CAAC;AAED,UAAW,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,IAAU,EAAE,EAAE;IACrC,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC;QAChB,KAAK,KAAK;YAAK,SAAS,CAAC,IAAI,CAAC,CAAC;YAAI,MAAM;QACzC,KAAK,KAAK;YAAK,SAAS,CAAC,IAAI,CAAC,CAAC;YAAI,MAAM;QACzC,KAAK,QAAQ;YAAE,YAAY,CAAC,IAAI,CAAC,CAAC;YAAC,MAAM;IAC7C,CAAC;IAED,OAAO,CAAC,KAAK,CAAC,SAAS,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC;IACtC,OAAO,CAAC,MAAM,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;IACpC,UAAW,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;AACpC,CAAC,CAAC,CAAC"}