@genai-fi/nanogpt 0.7.0 → 0.7.1

package/dist/{webgpu_util-g13LvDIv.js → webgpu_program-WaoMq-WD.js}

@@ -1,4 +1,4 @@
-import { aa as F, ab as O, ac as L, Z as N, l as _ } from "./index-C0dhsYom.js";
+import { aa as k, ab as z, ac as E, a1 as j, l as A } from "./index-UdZhlibC.js";
 /**
  * @license
  * Copyright 2019 Google LLC. All Rights Reserved.
@@ -15,16 +15,16 @@ import { aa as F, ab as O, ac as L, Z as N, l as _ } from "./index-C0dhsYom.js";
  * limitations under the License.
  * =============================================================================
  */
-function U(e, s) {
+function L(e, s) {
   if (Math.max(...e) > 5)
     throw new Error("Cannot symbolically compute strides for rank > 6 tensor.");
-  const t = e.length, o = "xyzwuv", n = e.map((a) => `${s}.${o[a]}`), r = new Array(t - 1);
-  r[t - 2] = n[t - 1];
-  for (let a = t - 3; a >= 0; --a)
-    r[a] = `(${r[a + 1]} * ${n[a + 1]})`;
-  return r;
+  const t = e.length, o = "xyzwuv", i = e.map((n) => `${s}.${o[n]}`), u = new Array(t - 1);
+  u[t - 2] = i[t - 1];
+  for (let n = t - 3; n >= 0; --n)
+    u[n] = `(${u[n + 1]} * ${i[n + 1]})`;
+  return u;
 }
-const H = (e, s, t) => t === "int32" ? `atomicAdd(${e}, bitcast<i32>(${s}));` : `
+const X = (e, s, t) => t === "int32" ? `atomicAdd(${e}, bitcast<i32>(${s}));` : `
           {
             var oldValue = 0;
             loop {
@@ -53,24 +53,24 @@ const H = (e, s, t) => t === "int32" ? `atomicAdd(${e}, bitcast<i32>(${s}));` :
  * limitations under the License.
  * =============================================================================
  */
-var y;
+var b;
 (function(e) {
   e[e.FROM_PIXELS = 0] = "FROM_PIXELS", e[e.DRAW = 1] = "DRAW";
-})(y || (y = {}));
-const Z = (e, s, t, o, n) => {
-  const r = { dtype: o.dtype, shape: o.shape }, a = P(t, r, s), i = e.createShaderModule({ code: a, label: s.constructor.name });
-  let d = L().get("WEBGPU_PRINT_SHADER");
+})(b || (b = {}));
+const H = (e, s, t, o, i) => {
+  const u = { dtype: o.dtype, shape: o.shape }, n = D(t, u, s), r = e.createShaderModule({ code: n, label: s.constructor.name });
+  let d = E().get("WEBGPU_PRINT_SHADER");
   if (d !== "") {
     d = d.toLowerCase();
     const p = d.split(",");
-    (d === "all" || p.some((u) => s.shaderKey.toLowerCase().includes(u))) && (console.group(s.shaderKey), console.debug(a), console.groupEnd());
+    (d === "all" || p.some((a) => s.shaderKey.toLowerCase().includes(a))) && (console.group(s.shaderKey), console.debug(n), console.groupEnd());
   }
-  return n ? e.createComputePipelineAsync({
-    compute: { module: i, entryPoint: "_start" },
+  return i ? e.createComputePipelineAsync({
+    compute: { module: r, entryPoint: "_start" },
     label: s.constructor.name,
     layout: "auto"
   }) : e.createComputePipeline({
-    compute: { module: i, entryPoint: "_start" },
+    compute: { module: r, entryPoint: "_start" },
     label: s.constructor.name,
     layout: "auto"
   });
@@ -118,7 +118,7 @@ function I(e) {
     return "v";
   throw Error(`Index ${e} is not yet supported`);
 }
-function q(...e) {
+function Y(...e) {
   let s;
   switch (e.length) {
     case 0:
@@ -139,7 +139,7 @@ function q(...e) {
 function w(e, s) {
   let t;
   return t = `
-     ${D(s)}
+     ${N(s)}
       fn _start(@builtin(local_invocation_id) LocalId : vec3<u32>,
                 @builtin(global_invocation_id) GlobalId : vec3<u32>,
                 @builtin(local_invocation_index) LocalIndex: u32,
@@ -154,13 +154,13 @@ function w(e, s) {
       }
     `, t;
 }
-function D(e) {
+function N(e) {
   return `
   @compute @workgroup_size(${e.workgroupSize[0]}, ${e.workgroupSize[1]}, ${e.workgroupSize[2]})
 `;
 }
-function P(e, s, t) {
-  const o = [], n = t.workgroupSize[0] * t.workgroupSize[1] * t.workgroupSize[2];
+function D(e, s, t) {
+  const o = [], i = t.workgroupSize[0] * t.workgroupSize[1] * t.workgroupSize[2];
   if (t.outputComponent = t.outputComponent ? t.outputComponent : 1, o.push(`
 
       var<private> localId: vec3<u32>;
@@ -171,13 +171,13 @@ function P(e, s, t) {
 
       // Only used when the y/z dimension of workgroup size is 1.
       fn getGlobalIndex() -> i32 {
-        ${E(t) ? "  return i32(globalId.x);" : `  return i32((workgroupId.z * numWorkgroups.x * numWorkgroups.y +
-                workgroupId.y * numWorkgroups.x + workgroupId.x) * ${n}u +
+        ${F(t) ? "  return i32(globalId.x);" : `  return i32((workgroupId.z * numWorkgroups.x * numWorkgroups.y +
+                workgroupId.y * numWorkgroups.x + workgroupId.x) * ${i}u +
                 localIndex);
         `}
       }
     `), t.pixelsOpType != null) {
-    const h = t.pixelsOpType === y.FROM_PIXELS ? `@group(0) @binding(0) var<storage, read_write> result: array<${b(s.dtype, t.outputComponent)}>;` : `@group(0) @binding(1) var<storage, read> inBuf : array<${b(e[0].dtype, t.outputComponent)}>;`, c = s.shape.length === 3 ? "vec2<i32>" : "i32";
+    const h = t.pixelsOpType === b.FROM_PIXELS ? `@group(0) @binding(0) var<storage, read_write> result: array<${x(s.dtype, t.outputComponent)}>;` : `@group(0) @binding(1) var<storage, read> inBuf : array<${x(e[0].dtype, t.outputComponent)}>;`, c = s.shape.length === 3 ? "vec2<i32>" : "i32";
     o.push(`
         struct Uniform {
           outShapeStrides : ${c},
@@ -189,63 +189,63 @@ function P(e, s, t) {
         ${h}
         @group(0) @binding(2) var<uniform> uniforms: Uniform;
       `);
-    const x = k(t);
+    const S = m(t);
     return [
       C,
       o.join(`
 `),
-      m(s.shape),
+      y(s.shape),
       t.getUserCode(),
-      w(x, t)
+      w(S, t)
     ].join(`
 `);
   }
-  let r, a, i = "struct Uniforms { NAN : f32, INFINITY : f32, ";
+  let u, n, r = "struct Uniforms { NAN : f32, INFINITY : f32, ";
   t.variableNames.forEach((h, c) => {
-    const x = v(e[c].shape.length);
-    i += `${h.charAt(0).toLowerCase() + h.slice(1)}Shape : ${x}, `, r = e[c].shape.length - 1, a = v(r), i += `${h.charAt(0).toLowerCase() + h.slice(1)}ShapeStrides: ${a}, `;
+    const S = v(e[c].shape.length);
+    r += `${h.charAt(0).toLowerCase() + h.slice(1)}Shape : ${S}, `, u = e[c].shape.length - 1, n = v(u), r += `${h.charAt(0).toLowerCase() + h.slice(1)}ShapeStrides: ${n}, `;
   });
   const d = v(s.shape.length);
-  i += `outShape : ${d}, `, r = s.shape.length - 1, a = v(r), i += `
-         outShapeStrides: ${a}, `, t.size && (i += "size : i32, "), t.uniforms && (i += t.uniforms), i += "};", i = K(i), o.push(i), t.atomic ? o.push(`
+  r += `outShape : ${d}, `, u = s.shape.length - 1, n = v(u), r += `
+         outShapeStrides: ${n}, `, t.size && (r += "size : i32, "), t.uniforms && (r += t.uniforms), r += "};", r = B(r), o.push(r), t.atomic ? o.push(`
       @group(0) @binding(0) var<storage, read_write> result: array<atomic<i32>>;
     `) : o.push(`
-      @group(0) @binding(0) var<storage, read_write> result: array<${b(s.dtype, t.outputComponent)}>;
+      @group(0) @binding(0) var<storage, read_write> result: array<${x(s.dtype, t.outputComponent)}>;
     `), t.variableNames.forEach((h, c) => {
     o.push(`
-      @group(0) @binding(${1 + c}) var<storage, read> ${h}: array<${t.variableComponents ? b(e[c].dtype, t.variableComponents[c]) : b(e[c].dtype, t.outputComponent)}>;
+      @group(0) @binding(${1 + c}) var<storage, read> ${h}: array<${t.variableComponents ? x(e[c].dtype, t.variableComponents[c]) : x(e[c].dtype, t.outputComponent)}>;
         `);
-  }), i !== "" && o.push(`
+  }), r !== "" && o.push(`
       @group(0) @binding(${1 + t.variableNames.length}) var<uniform> uniforms: Uniforms;
       `);
-  const p = T(s.shape, t.dispatchLayout), u = [
+  const p = R(s.shape, t.dispatchLayout), a = [
     C,
     o.join(`
-`) + W,
-    m(s.shape),
+`) + T,
+    y(s.shape),
     p,
-    V(s.shape.length)
+    G(s.shape.length)
   ];
-  t.atomic || u.push(B(s.shape, s.dtype, t.outputComponent)), t.variableNames.forEach((h, c) => {
-    u.push(`${m(e[c].shape, h)}`);
+  t.atomic || a.push(V(s.shape, s.dtype, t.outputComponent)), t.variableNames.forEach((h, c) => {
+    a.push(`${y(e[c].shape, h)}`);
   });
-  const g = e.map((h, c) => G(h, s.shape, t.variableComponents ? t.variableComponents[c] : t.outputComponent, t.dispatchLayout.x.length === s.shape.length)).join(`
+  const g = e.map((h, c) => P(h, s.shape, t.variableComponents ? t.variableComponents[c] : t.outputComponent, t.dispatchLayout.x.length === s.shape.length)).join(`
 `);
-  u.push(g), u.push(t.getUserCode());
-  const l = k(t);
-  return u.push(w(l, t)), u.join(`
+  a.push(g), a.push(t.getUserCode());
+  const l = m(t);
+  return a.push(w(l, t)), a.join(`
 `);
 }
-function J(e, s, t) {
+function q(e, s, t) {
   let o = e.shaderKey;
   if (e.pixelsOpType != null)
     return o;
-  const n = [], r = [];
-  s.forEach((u) => {
-    n.push(u.shape), r.push(u.dtype);
-  }), n.push(t.shape), r.push(t.dtype);
-  const a = s.map((u) => F(u.shape, t.shape)), i = s.map((u) => O(u.shape, t.shape)).join("_"), d = a.map((u) => u.join("_")).join(";"), p = E(e) ? "flatDispatch" : "";
-  return o += "_" + (e.workgroupSize ? e.workgroupSize.join(",") : "") + n.map((u) => u.length).join(",") + r.join(",") + e.variableNames.join(",") + d + i + p, o;
+  const i = [], u = [];
+  s.forEach((a) => {
+    i.push(a.shape), u.push(a.dtype);
+  }), i.push(t.shape), u.push(t.dtype);
+  const n = s.map((a) => k(a.shape, t.shape)), r = s.map((a) => z(a.shape, t.shape)).join("_"), d = n.map((a) => a.join("_")).join(";"), p = F(e) ? "flatDispatch" : "";
+  return o += "_" + (e.workgroupSize ? e.workgroupSize.join(",") : "") + i.map((a) => a.length).join(",") + u.join(",") + e.variableNames.join(",") + d + r + p, o;
 }
 const C = `
   struct vec5 {x: i32, y: i32, z: i32, w: i32, u: i32};
@@ -297,138 +297,138 @@ const C = `
     let floatToUint: vec4<u32> = bitcast<vec4<u32>>(val);
     return (floatToUint & vec4<u32>(0x7fffffffu)) > vec4<u32>(0x7f800000u);
   }
-`, W = `
+`, T = `
   fn isinf(val: f32) -> bool {
     return abs(val) == uniforms.INFINITY;
   }
 `;
-function m(e, s = "") {
-  const t = e.length, o = s !== "" ? `get${s.charAt(0).toUpperCase() + s.slice(1)}CoordsFromIndex` : "getCoordsFromIndex", n = s !== "" ? `${s.charAt(0).toLowerCase() + s.slice(1)}ShapeStrides` : "outShapeStrides";
+function y(e, s = "") {
+  const t = e.length, o = s !== "" ? `get${s.charAt(0).toUpperCase() + s.slice(1)}CoordsFromIndex` : "getCoordsFromIndex", i = s !== "" ? `${s.charAt(0).toLowerCase() + s.slice(1)}ShapeStrides` : "outShapeStrides";
   if (t <= 1)
     return `fn ${o}(index : i32) -> i32 { return index; }`;
-  const r = N(e), a = v(t), i = [];
+  const u = j(e), n = v(t), r = [];
   for (let p = 0; p < t; p++)
-    i.push(`d${p}`);
-  if (r.length === 1)
+    r.push(`d${p}`);
+  if (u.length === 1)
     return `    fn ${o}(index : i32) -> vec2<i32> {
-      let d0 = index / uniforms.${n}; let d1 = index - d0 * uniforms.${n};
+      let d0 = index / uniforms.${i}; let d1 = index - d0 * uniforms.${i};
       return vec2<i32>(d0, d1);
     }`;
   let d;
-  return d = "var index2 = index;" + r.map((p, u) => {
-    const g = `let ${i[u]} = index2 / uniforms.${n}.${I(u)}`, l = u === r.length - 1 ? `let ${i[u + 1]} = index2 - ${i[u]} * uniforms.${n}.${I(u)}` : `index2 = index2 - ${i[u]} * uniforms.${n}.${I(u)}`;
+  return d = "var index2 = index;" + u.map((p, a) => {
+    const g = `let ${r[a]} = index2 / uniforms.${i}.${I(a)}`, l = a === u.length - 1 ? `let ${r[a + 1]} = index2 - ${r[a]} * uniforms.${i}.${I(a)}` : `index2 = index2 - ${r[a]} * uniforms.${i}.${I(a)}`;
     return `${g}; ${l};`;
   }).join(""), `
-    fn ${o}(index : i32) -> ${a} {
+    fn ${o}(index : i32) -> ${n} {
       ${d}
-      return ${a}(${i.join(",")});
+      return ${n}(${r.join(",")});
     }
   `;
 }
-function M(e, s) {
-  const t = e.name, o = e.shape.length, n = v(o), r = "get" + t.charAt(0).toUpperCase() + t.slice(1), a = ["d0", "d1", "d2", "d3", "d4", "d5"].slice(0, o), i = a.map((u) => `${u} : i32`).join(", ");
+function U(e, s) {
+  const t = e.name, o = e.shape.length, i = v(o), u = "get" + t.charAt(0).toUpperCase() + t.slice(1), n = ["d0", "d1", "d2", "d3", "d4", "d5"].slice(0, o), r = n.map((a) => `${a} : i32`).join(", ");
   if (o < 1)
     return `
-      fn ${r}() -> ${f(s)} {
+      fn ${u}() -> ${f(s)} {
         return ${f(s)}(${t}[0]);
       }
     `;
   const d = `uniforms.${t.charAt(0).toLowerCase() + t.slice(1)}Shape`;
   let p = `${o}D`;
   return o === 0 && (p = "1D"), `
-    fn ${r}(${i}) -> ${f(s)} {
-      return ${f(s)}(${t}[getIndexFromCoords${p}(${n}(${a.join(",")}),
+    fn ${u}(${r}) -> ${f(s)} {
+      return ${f(s)}(${t}[getIndexFromCoords${p}(${i}(${n.join(",")}),
         ${d})${s === 1 ? "" : ` / ${s}`}]);
     }
    `;
 }
-function R(e, s, t, o) {
-  const n = e.name, r = n.charAt(0).toUpperCase() + n.slice(1), a = "get" + r + "ByOutput", i = e.shape.length, d = s.length, p = v(d);
-  if (O(e.shape, s) && o)
+function W(e, s, t, o) {
+  const i = e.name, u = i.charAt(0).toUpperCase() + i.slice(1), n = "get" + u + "ByOutput", r = e.shape.length, d = s.length, p = v(d);
+  if (z(e.shape, s) && o)
     return `
-    fn ${a}Index(globalIndex : i32) -> ${f(t)} {
-      return ${f(t)}(${n}[globalIndex]);
+    fn ${n}Index(globalIndex : i32) -> ${f(t)} {
+      return ${f(t)}(${i}[globalIndex]);
     }
 
-    fn ${a}Coords(coords : ${p}) -> ${f(t)} {
-      return ${f(t)}(${n}[${d > 1 ? "getOutputIndexFromCoords(coords)" : "coords"}${t === 1 ? "" : ` / ${t}`}]);
+    fn ${n}Coords(coords : ${p}) -> ${f(t)} {
+      return ${f(t)}(${i}[${d > 1 ? "getOutputIndexFromCoords(coords)" : "coords"}${t === 1 ? "" : ` / ${t}`}]);
     }
     `;
-  const u = F(e.shape, s), g = d - i;
+  const a = k(e.shape, s), g = d - r;
   let l = "";
-  if (i === 0)
+  if (r === 0)
     return `
-    fn ${a}Index(globalIndex : i32) -> ${f(t)}{
-      return get${r}();
+    fn ${n}Index(globalIndex : i32) -> ${f(t)}{
+      return get${u}();
     }
 
-    fn ${a}Coords(coords : ${p}) -> ${f(t)}{
-      return get${r}();
+    fn ${n}Coords(coords : ${p}) -> ${f(t)}{
+      return get${u}();
     }
   `;
-  d < 2 && u.length >= 1 ? l = "coords = 0;" : l = u.map((x) => `coords.${I(x + g)} = 0;`).join(`
+  d < 2 && a.length >= 1 ? l = "coords = 0;" : l = a.map((S) => `coords.${I(S + g)} = 0;`).join(`
 `);
   let $ = "";
-  if (d < 2 && i > 0)
+  if (d < 2 && r > 0)
     $ = "coords";
   else if (d > 1) {
-    const x = v(i), j = e.shape.map((X, A) => `coords.${I(A + g)}`).join(", ");
-    $ = `${x}(${j})`;
+    const S = v(r), O = e.shape.map((M, _) => `coords.${I(_ + g)}`).join(", ");
+    $ = `${S}(${O})`;
   } else
     $ = "coords";
-  const h = `uniforms.${n.charAt(0).toLowerCase() + n.slice(1)}Shape`, c = `${i}D`;
+  const h = `uniforms.${i.charAt(0).toLowerCase() + i.slice(1)}Shape`, c = `${r}D`;
   return `
-  fn ${a}Index(globalIndex : i32) -> ${f(t)} {
+  fn ${n}Index(globalIndex : i32) -> ${f(t)} {
     var coords = getCoordsFromIndex(globalIndex);
     ${l}
-    return ${f(t)}(${n}[getIndexFromCoords${c}(${$}, ${h})${t === 1 ? "" : ` / ${t}`}]);
+    return ${f(t)}(${i}[getIndexFromCoords${c}(${$}, ${h})${t === 1 ? "" : ` / ${t}`}]);
   }
 
-  fn ${a}Coords(coordsIn : ${p}) -> ${f(t)} {
+  fn ${n}Coords(coordsIn : ${p}) -> ${f(t)} {
     var coords = coordsIn;
     ${l}
-    return ${f(t)}(${n}[getIndexFromCoords${c}(${$}, ${h})${t === 1 ? "" : ` / ${t}`}]);
+    return ${f(t)}(${i}[getIndexFromCoords${c}(${$}, ${h})${t === 1 ? "" : ` / ${t}`}]);
   }
 `;
 }
-function G(e, s, t, o) {
-  let n = M(e, t);
-  return e.shape.length <= s.length && (n += R(e, s, t, o)), n;
+function P(e, s, t, o) {
+  let i = U(e, t);
+  return e.shape.length <= s.length && (i += W(e, s, t, o)), i;
 }
-function T(e, s) {
-  const { x: t, y: o = [], z: n = [] } = s, r = e.length, a = t.length + o.length + n.length;
-  if (a !== r)
+function R(e, s) {
+  const { x: t, y: o = [], z: i = [] } = s, u = e.length, n = t.length + o.length + i.length;
+  if (n !== u)
     return "";
-  if (t.length === r)
-    return `fn getOutputCoords() -> ${v(r)}{
+  if (t.length === u)
+    return `fn getOutputCoords() -> ${v(u)}{
     let globalIndex = getGlobalIndex();
     return getCoordsFromIndex(globalIndex);
   }
   `;
-  let i = "";
-  const d = [t, o, n];
+  let r = "";
+  const d = [t, o, i];
   for (let l = 0; l < d.length; l++) {
     const $ = d[l];
     if ($.length !== 0)
       if ($.length === 1)
-        i += `let d${$[0]} = i32(globalId[${l}]);`;
+        r += `let d${$[0]} = i32(globalId[${l}]);`;
       else {
-        const h = U($, "uniforms.outShape");
-        i += `var index${l} = i32(globalId[${l}]);`;
+        const h = L($, "uniforms.outShape");
+        r += `var index${l} = i32(globalId[${l}]);`;
         for (let c = 0; c < h.length; c++)
-          i += `let d${$[c]} = index${l} / ${h[c]};`, c === h.length - 1 ? i += `let d${$[c + 1]} = index${l} - d${$[c]} * ${h[c]};` : i += `index${l} = index${l} - d${$[c]} * ${h[c]};`;
+          r += `let d${$[c]} = index${l} / ${h[c]};`, c === h.length - 1 ? r += `let d${$[c + 1]} = index${l} - d${$[c]} * ${h[c]};` : r += `index${l} = index${l} - d${$[c]} * ${h[c]};`;
       }
   }
   const p = [];
-  for (let l = 0; l < a; l++)
+  for (let l = 0; l < n; l++)
     p.push(`d${l}`);
-  const u = v(a);
-  let g = `fn getOutputCoords() -> ${u} {
-  ${i}
+  const a = v(n);
+  let g = `fn getOutputCoords() -> ${a} {
+  ${r}
 `;
-  return p.length === 0 ? g += `return ${u}(0); }` : g += `return ${u}(${p.join(",")}); }`, g;
+  return p.length === 0 ? g += `return ${a}(0); }` : g += `return ${a}(${p.join(",")}); }`, g;
 }
-function V(e) {
+function G(e) {
   let s = "";
   switch (e) {
     case 0:
@@ -485,141 +485,64 @@ function V(e) {
         `;
       break;
     default:
-      _(!1, () => `Unsupported ${e}D shape`);
+      A(!1, () => `Unsupported ${e}D shape`);
       break;
   }
   return s;
 }
-function E(e) {
+function F(e) {
   return e.dispatch[1] === 1 && e.dispatch[2] === 1;
 }
-function b(e, s = 1) {
+function x(e, s = 1) {
   if (e === "float32")
     return f(s, "f32");
   if (e === "int32" || e === "bool")
     return f(s, "i32");
   throw new Error(`type ${e} is not supported.`);
 }
-function B(e, s, t) {
-  const o = e.length, n = b(s, t);
-  let r = `fn setOutputAtIndex(flatIndex : i32, value : ${f(t)}) {
-      result[flatIndex] = ${n}(value);
+function V(e, s, t) {
+  const o = e.length, i = x(s, t);
+  let u = `fn setOutputAtIndex(flatIndex : i32, value : ${f(t)}) {
+      result[flatIndex] = ${i}(value);
     }
 
     fn setOutputAtIndexI32(flatIndex : i32, value : ${f(t, "i32")}) {
-      result[flatIndex] = ${n}(value);
+      result[flatIndex] = ${i}(value);
     }
     `;
   if (o >= 2) {
-    const a = ["d0", "d1", "d2", "d3", "d4", "d5"].slice(0, o), i = v(o);
-    r += `
-      fn setOutputAtCoords(${a.map((d) => `${d} : i32`).join(", ")}, value : ${f(t)}) {
-        let flatIndex = getOutputIndexFromCoords(${i}(${a.join(", ")}));
+    const n = ["d0", "d1", "d2", "d3", "d4", "d5"].slice(0, o), r = v(o);
+    u += `
+      fn setOutputAtCoords(${n.map((d) => `${d} : i32`).join(", ")}, value : ${f(t)}) {
+        let flatIndex = getOutputIndexFromCoords(${r}(${n.join(", ")}));
         setOutputAtIndex(flatIndex${t === 1 ? "" : ` / ${t}`}, value);
       }
-      fn setOutputAtCoordsI32(${a.map((d) => `${d} : i32`).join(", ")}, value : ${f(t, "i32")}) {
-        let flatIndex = getOutputIndexFromCoords(${i}(${a.join(", ")}));
+      fn setOutputAtCoordsI32(${n.map((d) => `${d} : i32`).join(", ")}, value : ${f(t, "i32")}) {
+        let flatIndex = getOutputIndexFromCoords(${r}(${n.join(", ")}));
         setOutputAtIndexI32(flatIndex${t === 1 ? "" : ` / ${t}`}, value);
       }
     `;
   }
-  return r;
+  return u;
 }
-function K(e) {
+function B(e) {
   const s = /(\w+)\s*:\s*vec(5|6)/g;
   e = e.replace(s, (o) => "@align(16) " + o);
   const t = /vec(5|6)\s*,\s*(\w+)/g;
-  return e = e.replace(t, (o, n, r) => `vec${n}, @align(16) ${r}`), e;
+  return e = e.replace(t, (o, i, u) => `vec${i}, @align(16) ${u}`), e;
 }
-function k(e) {
+function m(e) {
   return !(e.dispatchLayout.hasOwnProperty("y") && e.dispatchLayout.y.length !== 0 || e.dispatchLayout.hasOwnProperty("z") && e.dispatchLayout.z.length !== 0);
 }
-/**
- * @license
- * Copyright 2019 Google LLC. All Rights Reserved.
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- * =============================================================================
- */
-const S = (e) => {
-  let s = 1;
-  for (let t = 0; t < e.length; t++)
-    s *= e[t];
-  return s;
-};
-function Q(e, s, t = [1, 1, 1], o = [1, 1, 1]) {
-  const [n, r, a] = [
-    Math.ceil(S(e.x.map((i) => s[i])) / (t[0] * o[0])),
-    e.y ? Math.ceil(S(e.y.map((i) => s[i])) / (t[1] * o[1])) : 1,
-    e.z ? Math.ceil(S(e.z.map((i) => s[i])) / (t[2] * o[2])) : 1
-  ];
-  return [n, r, a];
-}
-function ee(e, s, t, o = !1) {
-  const n = [8, 8, 1], r = [4, 4, 1];
-  return o || (e <= 8 && (r[1] = 1), s <= 16 && t <= 16 && (n[0] = 4)), { workgroupSize: n, elementsPerThread: r };
-}
-function te(e, s, t = !1) {
-  if (t)
-    return [8, 8, 1];
-  const o = S(e.x.map((r) => s[r])), n = S(e.y.map((r) => s[r]));
-  return o <= 4 ? [4, 16, 1] : n <= 4 ? [16, 4, 1] : [16, 16, 1];
-}
-function se(e, s, t = !1) {
-  if (t)
-    return [4, 4, 1];
-  const o = S(e.x.map((r) => s[r])), n = S(e.y.map((r) => s[r]));
-  return o <= 4 ? [1, 2, 1] : n <= 4 ? [2, 1, 1] : [2, 2, 1];
-}
-function oe(e) {
-  return { x: e.map((s, t) => t) };
-}
-function re(e) {
-  if (e === "float32" || e === "int32" || e === "bool" || e === "string")
-    return 4;
-  if (e === "complex64")
-    return 8;
-  throw new Error(`Unknown dtype ${e}`);
-}
-function ne() {
-  return !!(typeof globalThis < "u" && globalThis.navigator && globalThis.navigator.gpu);
-}
-function ie(e, s) {
-  Array.isArray(e) || (e = [e]), e.forEach((t) => {
-    t != null && _(t.dtype !== "complex64", () => `${s} does not support complex64 tensors in the WebGPU backend.`);
-  });
-}
-var z;
-(function(e) {
-  e[e.MatMulReduceProgram = 0] = "MatMulReduceProgram", e[e.MatMulSplitKProgram = 1] = "MatMulSplitKProgram", e[e.MatMulSmallOutputSizeProgram = 2] = "MatMulSmallOutputSizeProgram", e[e.MatMulPackedProgram = 3] = "MatMulPackedProgram", e[e.MatMulMax = 4] = "MatMulMax";
-})(z || (z = {}));
 export {
-  re as G,
-  z as M,
-  y as P,
-  Z as a,
-  ee as b,
-  Q as c,
-  H as d,
-  v as e,
-  oe as f,
-  q as g,
-  I as h,
-  ne as i,
-  ie as j,
-  te as k,
-  se as l,
-  J as m,
-  b as n,
-  m as o,
+  b as P,
+  X as a,
+  v as b,
+  H as c,
+  I as d,
+  x as e,
+  y as f,
+  Y as g,
+  q as m,
   f as t
 };

package/dist/webgpu_util-DhSeP4b6.js

@@ -0,0 +1,80 @@
+import { l as u } from "./index-UdZhlibC.js";
+/**
+ * @license
+ * Copyright 2019 Google LLC. All Rights Reserved.
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ * =============================================================================
+ */
+const e = (r) => {
+  let t = 1;
+  for (let n = 0; n < r.length; n++)
+    t *= r[n];
+  return t;
+};
+function m(r, t, n = [1, 1, 1], a = [1, 1, 1]) {
+  const [o, i, f] = [
+    Math.ceil(e(r.x.map((c) => t[c])) / (n[0] * a[0])),
+    r.y ? Math.ceil(e(r.y.map((c) => t[c])) / (n[1] * a[1])) : 1,
+    r.z ? Math.ceil(e(r.z.map((c) => t[c])) / (n[2] * a[2])) : 1
+  ];
+  return [o, i, f];
+}
+function d(r, t, n, a = !1) {
+  const o = [8, 8, 1], i = [4, 4, 1];
+  return a || (r <= 8 && (i[1] = 1), t <= 16 && n <= 16 && (o[0] = 4)), { workgroupSize: o, elementsPerThread: i };
+}
+function p(r, t, n = !1) {
+  if (n)
+    return [8, 8, 1];
+  const a = e(r.x.map((i) => t[i])), o = e(r.y.map((i) => t[i]));
+  return a <= 4 ? [4, 16, 1] : o <= 4 ? [16, 4, 1] : [16, 16, 1];
+}
+function M(r, t, n = !1) {
+  if (n)
+    return [4, 4, 1];
+  const a = e(r.x.map((i) => t[i])), o = e(r.y.map((i) => t[i]));
+  return a <= 4 ? [1, 2, 1] : o <= 4 ? [2, 1, 1] : [2, 2, 1];
+}
+function h(r) {
+  return { x: r.map((t, n) => n) };
+}
+function x(r) {
+  if (r === "float32" || r === "int32" || r === "bool" || r === "string")
+    return 4;
+  if (r === "complex64")
+    return 8;
+  throw new Error(`Unknown dtype ${r}`);
+}
+function g() {
+  return !!(typeof globalThis < "u" && globalThis.navigator && globalThis.navigator.gpu);
+}
+function b(r, t) {
+  Array.isArray(r) || (r = [r]), r.forEach((n) => {
+    n != null && u(n.dtype !== "complex64", () => `${t} does not support complex64 tensors in the WebGPU backend.`);
+  });
+}
+var s;
+(function(r) {
+  r[r.MatMulReduceProgram = 0] = "MatMulReduceProgram", r[r.MatMulSplitKProgram = 1] = "MatMulSplitKProgram", r[r.MatMulSmallOutputSizeProgram = 2] = "MatMulSmallOutputSizeProgram", r[r.MatMulPackedProgram = 3] = "MatMulPackedProgram", r[r.MatMulMax = 4] = "MatMulMax";
+})(s || (s = {}));
+export {
+  x as G,
+  s as M,
+  d as a,
+  b,
+  m as c,
+  p as d,
+  M as e,
+  h as f,
+  g as i
+};